CUDA and Triton implementations of Flash Attention with SoftmaxN.

Official Github repository for the SIGCOMM '24 paper "Accelerating Model Training in Multi-cluster Environments with Consumer-grade GPUs"

Post-Training Quantization for Vision transformers.

Official repository for the Boltz biomolecular interaction models

AlphaFold 3 Input Generation with MMseqs2 and plmMSA