TSAIL group
- 961 followers
- FIT Building, Tsinghua University, Beijing, China
- https://ml.cs.tsinghua.edu.cn
Pinned Loading
Repositories
- SpargeAttn Public
SpargeAttention: A training-free sparse attention that can accelerate any model inference.
thu-ml/SpargeAttn’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 - 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 - MMTrustEval Public
A toolbox for benchmarking trustworthiness of multimodal large language models (MultiTrust, NeurIPS 2024 Track Datasets and Benchmarks)
thu-ml/MMTrustEval’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 - DiffusionBridge Public
Official codebase for "Diffusion Bridge Implicit Models" (ICLR 2025) and "Consistency Diffusion Bridge Models" (NeurIPS 2024)
thu-ml/DiffusionBridge’s past year of commit activity