TSAIL group

3D_Corruptions_AD

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

ares

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

Bridge-TTS

Official codebase for "Schrodinger Bridges Beat Diffusion Models on Text-to-Speech Synthesis" (https://arxiv.org/abs/2312.03491).

controlvideo

Official implementation for "ControlVideo: Adding Conditional Control for One Shot Text-to-Video Editing"

CRM

Single Image to 3D Textured Mesh in 10 seconds with Convolutional Reconstruction Model.

prolificdreamer

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

tianshou

An elegant PyTorch deep reinforcement learning library.

unidiffuser

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

warplda

Cache efficient implementation for Latent Dirichlet Allocation

zhusuan

A probabilistic programming library for Bayesian deep learning, generative models, based on Tensorflow

thu-ml/tianshou

An elegant PyTorch deep reinforcement learning library.

thu-ml/prolificdreamer

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

thu-ml/unidiffuser

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

thu-ml/ares

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

thu-ml/CRM

Single Image to 3D Textured Mesh in 10 seconds with Convolutional Reconstruction Model.

thu-ml/controlvideo

Official implementation for "ControlVideo: Adding Conditional Control for One Shot Text-to-Video Editing"

thu-ml/Bridge-TTS

Official codebase for "Schrodinger Bridges Beat Diffusion Models on Text-to-Speech Synthesis" (https://arxiv.org/abs/2312.03491).

thu-ml/3D_Corruptions_AD

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

thu-ml/low-bit-optimizers

Low-bit optimizers for PyTorch

thu-ml/DPM-Solver-v3

Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)

thu-ml/Attack-Bard

thu-ml/zh-clip

thu-ml/HiDe-Prompt

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

thu-ml/GNOT

thu-ml/Prior-Guided-RGF

thu-ml/AT3D

Towards Effective Adversarial Textured 3D Meshes on Physical Face Recognition, CVPR 2023, Highlight

thu-ml/SRPO

Codes accompanying the paper "Score Regularized Policy Optimization through Diffusion Behavior" (ICLR 2024).

thu-ml/DPOT

Code for "DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training"

thu-ml/IODF

thu-ml/NUNO

[ICML 2023] Non-Uniform Neural Operator (NUNO)

thu-ml/CEP-energy-guided-diffusion

Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction

thu-ml/Noise-Contrastive-Alignment

Code accompanying the paper "Noise Contrastive Alignment of Language Models with Explicit Rewards"

thu-ml/i-DODE

Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)

thu-ml/2by4-pretrain-acc-examples

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

thu-ml/ProbML-book-solution

thu-ml/adversarial_training_imagenet

thu-ml/VCAS

Official code for "Efficient Backpropagation with Variance Controlled Adaptive Sampling" (ICLR 2024)

thu-ml/imagenet-a-plus

thu-ml/ACTNN-PaddlePaddle

thu-ml/PINNacle

Codebase for PINNacle: A Comprehensive Benchmark of Physics-Informed Neural Networks for Solving PDEs.