Tsinghua Machine Learning Group (@thu-ml)
Guthub

thu-ml

Twitter logo Share LinkedIn logo Share Facebook logo Share

Top repositories

1

tianshou

An elegant PyTorch deep reinforcement learning library.
Python
7,267
star
2

zhusuan

A probabilistic programming library for Bayesian deep learning, generative models, based on Tensorflow
Python
2,200
star
3

prolificdreamer

Official code of ProlificDreamer: High-Fidelity and Diverse Text-to-3D Generation with Variational Score Distillation (NeurIPS 2023 Spotlight)
Python
1,358
star
4

unidiffuser

Code and models for the paper "One Transformer Fits All Distributions in Multi-Modal Diffusion"
Python
1,267
star
5

ares

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

CRM

Single Image to 3D Textured Mesh in 10 seconds with Convolutional Reconstruction Model.
Python
253
star
7

controlvideo

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

warplda

Cache efficient implementation for Latent Dirichlet Allocation
C++
160
star
9

3D_Corruptions_AD

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

low-bit-optimizers

Low-bit optimizers for PyTorch
Python
88
star
11

stochastic_gcn

Stochastic training of graph convolutional networks
Python
84
star
12

Attack-Bard

Python
53
star
13

DPM-Solver-v3

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

tianshou-docs-zh_CN

天授中文文档
TeX
46
star
15

Prior-Guided-RGF

Python
41
star
16

zh-clip

Python
41
star
17

vflow

Official code for "VFlow: More Expressive Generative Flows with Variational Data Augmentation" (ICML 2020)
Python
35
star
18

implicit-normalizing-flows

Code for "Implicit Normalizing Flows" (ICLR 2021 spotlight)
Python
34
star
19

HiDe-Prompt

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

AT3D

Towards Effective Adversarial Textured 3D Meshes on Physical Face Recognition, CVPR 2023, Highlight
Python
29
star
21

BigTopicModel

Big Topic Model is a fast engine for running large-scale Topic Models.
C++
22
star
22

IODF

C++
16
star
23

CF-UIcA

Code for "Collaborative Filtering with User-Item Co-Autoregressive Models"
Python
15
star
24

fpovi

Code for "Function Space Particle Optimization for Bayesian Neural Networks"
Python
15
star
25

NUNO

[ICML 2023] Non-Uniform Neural Operator (NUNO)
Python
15
star
26

Zhusuan-Jittor

Zhusuan with backend Jittor
Python
14
star
27

LM-Calibration

Python
12
star
28

mmdcgm-ssl

mmDCGMs for accurate classification and excellent class-conditional generation in semi-supervised learning
Python
11
star
29

SRPO

Codes accompanying the paper "Score Regularized Policy Optimization through Diffusion Behavior"
Python
11
star
30

Zhusuan-PaddlePaddle

Zhusuan with backend PaddlePaddle
Python
8
star
31

ood-dgm

Python
8
star
32

MEM_DGM

Code for "Learning to Generate with Memory"
Python
8
star
33

ProbML-book-solution

Jupyter Notebook
7
star
34

pmd

Population matching discrepancy
Python
7
star
35

adversarial_training_imagenet

Python
6
star
36

VCAS

Official code for "Efficient Backpropagation with Variance Controlled Adaptive Sampling" (ICLR 2024)
Python
6
star
37

imagenet-a-plus

Python
4
star
38

wmvl

Code for "A Wasserstein Minimum Velocity Approach to Learning Unnormalized Models"
Jupyter Notebook
3
star
39

sEM-vr

code for pLSA and LDA in the paper "Stochastic Expectation Maximization with Variance Reduction"
C++
3
star
40

Noise-Contrastive-Alignment

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

i-DODE

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

ACTNN-PaddlePaddle

Python
1
star