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 421

  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.9k 144

  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 510 87

  6. tianshou tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    Python 8.6k 1.2k

Repositories

Showing 10 of 77 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,904 Apache-2.0 144 65 9 Updated Jul 5, 2025
  • tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    thu-ml/tianshou’s past year of commit activity
    Python 8,611 MIT 1,163 144 (1 issue needs help) 8 Updated Jul 3, 2025
  • Adaptive-Sparse-Trainer Public

    Official implementation for "Pruning Large Language Models with Semi-Structural Adaptive Sparse Training" (AAAI 2025)

    thu-ml/Adaptive-Sparse-Trainer’s past year of commit activity
    Python 12 Apache-2.0 1 0 0 Updated Jul 1, 2025
  • 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 41 MIT 3 0 0 Updated Jun 30, 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 152 CC-BY-SA-4.0 10 1 0 Updated Jun 27, 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 16 Apache-2.0 1 0 0 Updated Jun 20, 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 635 Apache-2.0 45 24 4 Updated Jun 19, 2025
  • DDO Public
    thu-ml/DDO’s past year of commit activity
    Python 0 0 0 0 Updated Jun 16, 2025
  • GFT Public
    thu-ml/GFT’s past year of commit activity
    Python 34 MIT 0 4 0 Updated Jun 13, 2025
  • SparseDM Public
    thu-ml/SparseDM’s past year of commit activity
    Python 1 0 0 0 Updated Jun 10, 2025

Most used topics

Loading…