TSAIL group
Tsinghua Statistical Artificial Intelligence & Learning Group
- 886 followers
- FIT Building, Tsinghua University, Beijing, China
- https://ml.cs.tsinghua.edu.cn
Pinned Loading
Repositories
Showing 10 of 72 repositories
- SpargeAttn Public
SpargeAttention: A training-free sparse attention that can accelerate any model inference.
- DiffusionBridge Public
Official codebase for "Diffusion Bridge Implicit Models" (ICLR 2025) and "Consistency Diffusion Bridge Models" (NeurIPS 2024)
- MMTrustEval Public
A toolbox for benchmarking trustworthiness of multimodal large language models (MultiTrust, NeurIPS 2024 Track Datasets and Benchmarks)
- TetraJet-MXFP4Training Public
Pytorch implementation of "Oscillation-Reduced MXFP4 Training for Vision Transformers" on DeiT Model Pre-training
- SageAttention Public
Quantized Attention that achieves speedups of 2.1-3.1x and 2.7-5.1x compared to FlashAttention2 and xformers, respectively, without lossing end-to-end metrics across various models.