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

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

    Python 484 88

  3. unidiffuser unidiffuser Public

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

    Python 1.4k 86

  4. prolificdreamer prolificdreamer Public

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

    Python 1.5k 44

Repositories

Showing 10 of 62 repositories
  • HiDe-PET Public
    thu-ml/HiDe-PET’s past year of commit activity
    Python 0 MIT 0 0 0 Updated Nov 15, 2024
  • HiDe-Prompt Public

    Hierarchical Decomposition of Prompt-Based Continual Learning: Rethinking Obscured Sub-optimality (NeurIPS 2023, Spotlight)

    thu-ml/HiDe-Prompt’s past year of commit activity
    Python 74 MIT 7 3 0 Updated Nov 15, 2024
  • SageAttention Public

    Quantized Attention that achieves speedups of 2.1x and 2.7x compared to FlashAttention2 and xformers, respectively, without lossing end-to-end metrics across various models.

    thu-ml/SageAttention’s past year of commit activity
    Python 391 BSD-3-Clause 16 12 0 Updated Nov 14, 2024
  • RoboticsDiffusionTransformer Public

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

    thu-ml/RoboticsDiffusionTransformer’s past year of commit activity
    Python 451 MIT 37 8 1 Updated Nov 13, 2024
  • Noise-Contrastive-Alignment Public

    Code accompanying the paper "Noise Contrastive Alignment of Language Models with Explicit Rewards" (NeurIPS 2024)

    thu-ml/Noise-Contrastive-Alignment’s past year of commit activity
    Python 32 MIT 3 0 0 Updated Nov 7, 2024
  • 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 107 CC-BY-SA-4.0 7 4 0 Updated Nov 5, 2024
  • Efficient-Diffusion-Alignment Public

    Official Codebase for "Aligning Diffusion Behaviors with Q-functions for Efficient Continuous Control" (NeurIPS 2024)

    thu-ml/Efficient-Diffusion-Alignment’s past year of commit activity
    Python 9 0 0 0 Updated Oct 29, 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 17 MIT 0 0 0 Updated Oct 28, 2024
  • tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    thu-ml/tianshou’s past year of commit activity
    Python 7,962 MIT 1,126 136 (1 issue needs help) 3 Updated Oct 26, 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
    1 0 1 0 Updated Oct 25, 2024
View all repositories