thu-ml/SageAttention
[ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of 2-5x compared to FlashAttention, without losing end-to-end metrics across language, image, and video models.
Apache-2.0Cuda
- Stars
- 4.0k
- Forks
- 534
- Open issues
- 178
- Last push
- 9mo ago