Skip to content
@thu-ml

TSAIL group

Tsinghua Statistical Artificial Intelligence & Learning Group

Pinned Loading

  1. TurboDiffusion TurboDiffusion Public

    TurboDiffusion: 100–200× Acceleration for Video Diffusion Models

    Python 3.3k 221

  2. unidiffuser unidiffuser Public

    Code and models for the paper "One Transformer Fits All Distributions in Multi-Modal Diffusion"

    Python 1.5k 92

  3. SageAttention SageAttention Public

    [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.

    Cuda 3.1k 327

  4. prolificdreamer prolificdreamer Public

    ProlificDreamer: High-Fidelity and Diverse Text-to-3D Generation with Variational Score Distillation (NeurIPS 2023 Spotlight)

    Python 1.6k 47

  5. ares ares Public

    A Python library for adversarial machine learning focusing on benchmarking adversarial robustness.

    Python 524 93

  6. tianshou tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    Python 10k 1.3k

Repositories

Showing 10 of 87 repositories
  • TurboDiffusion Public

    TurboDiffusion: 100–200× Acceleration for Video Diffusion Models

    thu-ml/TurboDiffusion’s past year of commit activity
    Python 3,285 Apache-2.0 221 57 6 Updated Jan 29, 2026
  • i-DODE Public

    Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)

    thu-ml/i-DODE’s past year of commit activity
    Python 18 Apache-2.0 1 1 0 Updated Jan 27, 2026
  • DiT-Extrapolation Public

    Official implementation for "RIFLEx: A Free Lunch for Length Extrapolation in Video Diffusion Transformers" (ICML 2025) and UltraViCo (ICLR 2026)

    thu-ml/DiT-Extrapolation’s past year of commit activity
    Python 780 Apache-2.0 74 23 0 Updated Jan 26, 2026
  • embodied-data-toolkit Public

    A toolkit for processing raw embodied data into standardized formats and converting between embodied dataset schemas.

    thu-ml/embodied-data-toolkit’s past year of commit activity
    Python 6 1 0 0 Updated Jan 21, 2026
  • RoboticsDiffusionTransformer Public

    RDT-1B: a Diffusion Foundation Model for Bimanual Manipulation

    thu-ml/RoboticsDiffusionTransformer’s past year of commit activity
    Python 1,605 MIT 150 36 3 Updated Jan 21, 2026
  • SLA Public

    SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention

    thu-ml/SLA’s past year of commit activity
    Python 258 Apache-2.0 15 7 0 Updated Jan 17, 2026
  • SageAttention Public

    [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.

    thu-ml/SageAttention’s past year of commit activity
    Cuda 3,108 Apache-2.0 327 150 16 Updated Jan 18, 2026
  • MLA-Trust Public

    A toolbox for benchmarking Multimodal LLM Agents trustworthiness across truthfulness, controllability, safety and privacy dimensions through 34 interactive tasks

    thu-ml/MLA-Trust’s past year of commit activity
    Python 62 MIT 4 2 0 Updated Jan 9, 2026
  • Motus Public

    Official code of Motus: A Unified Latent Action World Model

    thu-ml/Motus’s past year of commit activity
    Python 607 Apache-2.0 11 15 0 Updated Jan 5, 2026
  • SpargeAttn Public

    [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.

    thu-ml/SpargeAttn’s past year of commit activity
    Cuda 920 Apache-2.0 82 57 3 Updated Dec 31, 2025