TSAIL group
Tsinghua Statistical Artificial Intelligence & Learning Group
- 980 followers
- FIT Building, Tsinghua University, Beijing, China
- https://ml.cs.tsinghua.edu.cn
Pinned Loading
Repositories
Showing 10 of 74 repositories
- SpargeAttn Public
SpargeAttention: A training-free sparse attention that can accelerate any model inference.
thu-ml/SpargeAttn’s past year of commit activity - TetraJet-MXFP4Training Public
Pytorch implementation of "Oscillation-Reduced MXFP4 Training for Vision Transformers" on DeiT Model Pre-training
thu-ml/TetraJet-MXFP4Training’s past year of commit activity - SageAttention Public
Quantized Attention achieves speedup of 2-3x and 3-5x compared to FlashAttention and xformers, without lossing end-to-end metrics across language, image, and video models.
thu-ml/SageAttention’s past year of commit activity - RIFLEx Public
Official implementation for "RIFLEx: A Free Lunch for Length Extrapolation in Video Diffusion Transformers" (ICML 2025)
thu-ml/RIFLEx’s past year of commit activity - cond-image-leakage Public
Official implementation for "Identifying and Solving Conditional Image Leakage in Image-to-Video Diffusion Model" (NeurIPS 2024)
thu-ml/cond-image-leakage’s past year of commit activity