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. unidiffuser unidiffuser Public

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

    Python 1.4k 88

  3. prolificdreamer prolificdreamer Public

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

    Python 1.5k 45

  4. SageAttention 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.

    Cuda 1.1k 64

  5. ares ares Public

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

    Python 500 87

  6. tianshou tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    Python 8.3k 1.1k

Repositories

Showing 10 of 72 repositories
  • tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    thu-ml/tianshou’s past year of commit activity
    Python 8,252 MIT 1,136 145 (1 issue needs help) 5 Updated Mar 6, 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 233 Apache-2.0 7 7 0 Updated Mar 5, 2025
  • 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 17 Apache-2.0 1 1 0 Updated Mar 4, 2025
  • MMTrustEval Public

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

    thu-ml/MMTrustEval’s past year of commit activity
    Python 132 CC-BY-SA-4.0 8 3 0 Updated Mar 4, 2025
  • RIFLEx Public

    Official implementation for "RIFLEx: A Free Lunch for Length Extrapolation in Video Diffusion Transformers"

    thu-ml/RIFLEx’s past year of commit activity
    Python 296 Apache-2.0 28 8 0 Updated Mar 3, 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 6 Apache-2.0 1 0 0 Updated Mar 3, 2025
  • 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
    Cuda 1,071 Apache-2.0 64 35 1 Updated Feb 28, 2025
  • STAIR Public

    Official codebase for "STAIR: Improving Safety Alignment with Introspective Reasoning"

    thu-ml/STAIR’s past year of commit activity
    Python 23 MIT 1 0 0 Updated Feb 26, 2025
  • EffWRN-paddle Public
    thu-ml/EffWRN-paddle’s past year of commit activity
    Python 0 MIT 0 0 0 Updated Feb 24, 2025
  • thu-ml/RIFLEx.github.io’s past year of commit activity
    JavaScript 0 0 0 0 Updated Feb 21, 2025

Top languages

Loading…

Most used topics

Loading…