Skip to content
@thu-ml

TSAIL group

Tsinghua Statistical Artificial Intelligence & Learning Group

Pinned Loading

  1. zhusuan Public

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

    Python 2.2k 419

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

    Cuda 1.5k 103

  3. unidiffuser Public

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

    Python 1.4k 90

  4. prolificdreamer Public

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

    Python 1.5k 45

  5. ares Public

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

    Python 505 86

  6. tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    Python 8.5k 1.1k

Repositories

Showing 10 of 72 repositories
  • tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    Python 8,479 MIT 1,149 147 (1 issue needs help) 6 Updated May 13, 2025
  • RIFLEx Public

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

    Python 658 Apache-2.0 63 17 0 Updated May 13, 2025
  • SpargeAttn Public

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

    Cuda 539 Apache-2.0 34 18 3 Updated May 11, 2025
  • 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.

    Cuda 1,482 Apache-2.0 103 46 6 Updated May 2, 2025
  • cond-image-leakage Public

    Official implementation for "Identifying and Solving Conditional Image Leakage in Image-to-Video Diffusion Model" (NeurIPS 2024)

    251 Apache-2.0 22 8 0 Updated Apr 25, 2025
  • RoboticsDiffusionTransformer Public

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

    Python 1,175 MIT 121 26 4 Updated Apr 4, 2025
  • MMTrustEval Public

    A toolbox for benchmarking trustworthiness of multimodal large language models (MultiTrust, NeurIPS 2024 Track Datasets and Benchmarks)

    Python 145 CC-BY-SA-4.0 10 1 0 Updated Mar 26, 2025
  • TetraJet-MXFP4Training Public

    Pytorch implementation of "Oscillation-Reduced MXFP4 Training for Vision Transformers" on DeiT Model Pre-training

    Python 10 Apache-2.0 1 0 0 Updated Mar 22, 2025
  • DiffusionBridge Public

    Official codebase for "Diffusion Bridge Implicit Models" (ICLR 2025) and "Consistency Diffusion Bridge Models" (NeurIPS 2024)

    Python 53 4 2 0 Updated Mar 10, 2025
  • GFT Public
    Python 30 MIT 0 4 0 Updated Mar 8, 2025