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 418

  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 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 758 41

  5. ares ares Public

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

    Python 492 88

  6. tianshou tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    Python 8.1k 1.1k

Repositories

Showing 10 of 64 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 758 Apache-2.0 41 25 1 Updated Dec 28, 2024
  • RoboticsDiffusionTransformer Public

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

    thu-ml/RoboticsDiffusionTransformer’s past year of commit activity
    Python 714 MIT 60 10 2 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 5 Apache-2.0 0 0 0 Updated Dec 22, 2024
  • ReMoE Public

    Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.

    thu-ml/ReMoE’s past year of commit activity
    Python 32 0 1 0 Updated Dec 20, 2024
  • tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    thu-ml/tianshou’s past year of commit activity
    Python 8,106 MIT 1,126 138 (1 issue needs help) 3 Updated Dec 10, 2024
  • tianshou-docs-zh_CN Public

    天授中文文档

    thu-ml/tianshou-docs-zh_CN’s past year of commit activity
    TeX 55 11 0 0 Updated Dec 10, 2024
  • 2by4-pretrain-acc-examples Public

    Code for "Accelerating Transformer Pre-training with 2:4 Sparsity"

    thu-ml/2by4-pretrain-acc-examples’s past year of commit activity
    Jupyter Notebook 16 BSD-3-Clause 0 0 0 Updated Dec 8, 2024
  • CRM Public

    [ECCV 2024] Single Image to 3D Textured Mesh in 10 seconds with Convolutional Reconstruction Model.

    thu-ml/CRM’s past year of commit activity
    Python 605 MIT 50 24 1 Updated Nov 28, 2024
  • 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 21 MIT 0 0 0 Updated Nov 18, 2024
  • DiffusionBridge Public

    Official codebase for "Diffusion Bridge Implicit Models" (https://arxiv.org/abs/2405.15885).

    thu-ml/DiffusionBridge’s past year of commit activity
    Python 14 1 1 0 Updated Nov 18, 2024