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 482 89

  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 61 repositories
  • tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    thu-ml/tianshou’s past year of commit activity
    Python 7,902 MIT 1,122 135 (1 issue needs help) 3 Updated Oct 26, 2024
  • DBIM Public

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

    thu-ml/DBIM’s past year of commit activity
    1 0 1 0 Updated Oct 25, 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 317 BSD-3-Clause 13 14 0 Updated Oct 24, 2024
  • RoboticsDiffusionTransformer Public

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

    thu-ml/RoboticsDiffusionTransformer’s past year of commit activity
    Python 321 MIT 27 2 0 Updated Oct 23, 2024
  • cond-image-leakage Public

    Official implementation for "Identifying and Solving Conditional Image Leakage in Image-to-Video Diffusion Model" (NeurIPS 2024)

    thu-ml/cond-image-leakage’s past year of commit activity
    279 Apache-2.0 28 4 0 Updated Oct 22, 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 16 MIT 0 0 0 Updated Oct 16, 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 558 MIT 45 23 0 Updated Oct 14, 2024
  • CEURL Public

    Official implementation for "PEAC: Unsupervised Pre-training for Cross-Embodiment Reinforcement Learning" (NeurIPS 2024)

    thu-ml/CEURL’s past year of commit activity
    Python 9 MIT 0 0 0 Updated Oct 13, 2024
  • 3D_Corruptions_AD Public

    Benchmarking Robustness of 3D Object Detection to Common Corruptions in Autonomous Driving, CVPR 2023

    thu-ml/3D_Corruptions_AD’s past year of commit activity
    Python 120 MIT 15 7 0 Updated Oct 9, 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 95 CC-BY-SA-4.0 7 4 0 Updated Sep 28, 2024
View all repositories

Most used topics

Loading…