TurboPairFormer:优化三角注意力内核,加速蛋白质折叠模型训练
搞蛋白质结构预测的可以看看,这个 TurboPairFormer 把 AlphaFold3 式训练的三角注意力算子重写了一遍,比 Triton 后端快 1.73 倍,数值还更稳定。
搞蛋白质结构预测的可以看看,这个 TurboPairFormer 把 AlphaFold3 式训练的三角注意力算子重写了一遍,比 Triton 后端快 1.73 倍,数值还更稳定。
NVIDIA 开源了这个工具,让 AI 代理能直接调用分子模型做药物发现。用上它任务完成率翻倍还省 Token,做生物计算的同学可以试试。