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 419

  2. unidiffuser unidiffuser Public

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

    Python 1.4k 87

  3. prolificdreamer prolificdreamer Public

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

    Python 1.5k 44

  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 955 58

  5. ares ares Public

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

    Python 496 87

  6. tianshou tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    Python 8.2k 1.1k

Repositories

Showing 10 of 67 repositories
  • 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 955 Apache-2.0 58 29 0 Updated Feb 15, 2025
  • oddefense Public
    thu-ml/oddefense’s past year of commit activity
    Jupyter Notebook 0 MIT 0 0 0 Updated Feb 14, 2025
  • CCA Public

    Codes accompanying the paper "Toward Guidance-Free AR Visual Generation via Condition Contrastive Alignment"

    thu-ml/CCA’s past year of commit activity
    Python 25 MIT 0 0 0 Updated Feb 12, 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 127 CC-BY-SA-4.0 8 3 0 Updated Feb 11, 2025
  • STAIR Public

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

    thu-ml/STAIR’s past year of commit activity
    12 MIT 0 0 0 Updated Feb 9, 2025
  • DiffusionBridge Public

    Official codebase for "Diffusion Bridge Implicit Models" (ICLR 2025)

    thu-ml/DiffusionBridge’s past year of commit activity
    Python 22 2 1 0 Updated Feb 5, 2025
  • tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    thu-ml/tianshou’s past year of commit activity
    Python 8,196 MIT 1,128 143 (1 issue needs help) 5 Updated Jan 26, 2025
  • GFT Public
    thu-ml/GFT’s past year of commit activity
    22 MIT 0 2 0 Updated Jan 26, 2025
  • RoboticsDiffusionTransformer Public

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

    thu-ml/RoboticsDiffusionTransformer’s past year of commit activity
    Python 885 MIT 83 17 3 Updated Dec 24, 2024
  • 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 7 Apache-2.0 0 0 0 Updated Dec 22, 2024