fast_trimul

一个用于快速三角乘法更新的即插即用库

fast_trimul 是一个基于 CuTe DSL 的即插即用式硬件无关库，专为 AlphaFold3 系列模型提供融合的 Triangle Multiplicative Updates 算子优化，可直接替换原有实现。它能将显存占用缩减至原先的一半左右，并几乎消除激活数据，对短序列场景的推理速度提升尤为显著。