TSAIL group
Tsinghua Statistical Artificial Intelligence & Learning Group
- 750 followers
- FIT Building, Tsinghua University, Beijing, China
- https://ml.cs.tsinghua.edu.cn
Pinned Loading
Repositories
Showing 10 of 64 repositories
- 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.
thu-ml/SageAttention’s past year of commit activity - Adaptive-Sparse-Trainer Public
Official implementation for "Pruning Large Language Models with Semi-Structural Adaptive Sparse Training" (AAAI 2025)
thu-ml/Adaptive-Sparse-Trainer’s past year of commit activity - 2by4-pretrain-acc-examples Public
Code for "Accelerating Transformer Pre-training with 2:4 Sparsity"
thu-ml/2by4-pretrain-acc-examples’s past year of commit activity - DiffusionBridge Public
Official codebase for "Diffusion Bridge Implicit Models" (https://arxiv.org/abs/2405.15885).
thu-ml/DiffusionBridge’s past year of commit activity