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.4k 240

  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.2k 370

  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 526 94

  6. tianshou tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    Python 10.4k 1.3k

Repositories

Showing 10 of 89 repositories
  • 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 15 2 0 0 Updated Mar 16, 2026
  • Causal-Forcing Public

    Official codebase for "Causal Forcing: Autoregressive Diffusion Distillation Done Right for High-Quality Real-Time Interactive Video Generation"

    thu-ml/Causal-Forcing’s past year of commit activity
    Python 465 Apache-2.0 21 12 1 Updated Mar 15, 2026
  • 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
    258 Apache-2.0 22 8 0 Updated Mar 13, 2026
  • DiT-Extrapolation Public

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

    thu-ml/DiT-Extrapolation’s past year of commit activity
    Python 793 Apache-2.0 74 24 0 Updated Mar 8, 2026
  • TurboDiffusion Public

    TurboDiffusion: 100–200× Acceleration for Video Diffusion Models

    thu-ml/TurboDiffusion’s past year of commit activity
    Python 3,404 Apache-2.0 240 66 8 Updated Mar 6, 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 959 Apache-2.0 87 58 2 Updated Feb 25, 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 291 Apache-2.0 18 9 0 Updated Feb 24, 2026
  • thu-ml/CausalForcing.github.io’s past year of commit activity
    JavaScript 0 0 0 0 Updated Feb 7, 2026
  • RDT2 Public

    Official code of RDT 2

    thu-ml/RDT2’s past year of commit activity
    Python 738 Apache-2.0 45 16 2 Updated Feb 7, 2026
  • thu-ml/ultraimage.github.io’s past year of commit activity
    JavaScript 0 0 0 0 Updated Feb 6, 2026