fast_trimul
A drop-in library for fast Triangle Multiplicative Updates
What is fast_trimul?
A drop-in, hardware-agnostic library for Fused Triangle Multiplicative Updates across AlphaFold3 family models, powered by CuTe DSL. Uses roughly half the GPU memory with almost zero activation data at any sequence length. It is far faster for small sequences.
Screenshots
?
No comments yet. Be the first!