Skip to content
@thu-ml

TSAIL group

Tsinghua Statistical Artificial Intelligence & Learning Group

Pinned Loading

  1. zhusuan zhusuan Public

    A probabilistic programming library for Bayesian deep learning, generative models, based on Tensorflow

    Python 2.2k 420

  2. SageAttention SageAttention Public

    Quantized Attention achieves speedup of 2-5x and 3-11x compared to FlashAttention and xformers, without lossing end-to-end metrics across language, image, and video models.

    Cuda 1.6k 115

  3. unidiffuser unidiffuser Public

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

    Python 1.4k 89

  4. prolificdreamer prolificdreamer Public

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

    Python 1.5k 46

  5. ares ares Public

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

    Python 507 87

  6. tianshou tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    Python 8.5k 1.2k

Repositories

Showing 10 of 74 repositories
  • SageAttention Public

    Quantized Attention achieves speedup of 2-5x and 3-11x 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
    Cuda 1,646 Apache-2.0 115 52 7 Updated May 31, 2025
  • tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    thu-ml/tianshou’s past year of commit activity
    Python 8,545 MIT 1,158 146 (1 issue needs help) 9 Updated May 27, 2025
  • FrameBridge Public

    FrameBridge: Improving Image-to-Video Generation with Bridge Models

    thu-ml/FrameBridge’s past year of commit activity
    0 0 0 0 Updated May 26, 2025
  • GFT Public
    thu-ml/GFT’s past year of commit activity
    Python 31 MIT 0 4 0 Updated May 26, 2025
  • UniCardio Public

    This is the official code of UniCardio

    thu-ml/UniCardio’s past year of commit activity
    5 0 0 0 Updated May 25, 2025
  • SpargeAttn Public

    SpargeAttention: A training-free sparse attention that can accelerate any model inference.

    thu-ml/SpargeAttn’s past year of commit activity
    Cuda 579 Apache-2.0 40 19 3 Updated May 25, 2025
  • 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
    Python 13 Apache-2.0 1 0 0 Updated May 24, 2025
  • 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
    Python 673 Apache-2.0 66 17 0 Updated May 13, 2025
  • 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
    252 Apache-2.0 22 8 0 Updated Apr 26, 2025
  • RoboticsDiffusionTransformer Public

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

    thu-ml/RoboticsDiffusionTransformer’s past year of commit activity
    Python 1,221 MIT 125 26 5 Updated Apr 4, 2025