Official PyTorch implementation for "Large Language Diffusion Models"

Official implementation for "RIFLEx: A Free Lunch for Length Extrapolation in Video Diffusion Transformers" (ICML 2025) , UltraViCo (ICLR 2026) and UltraImage

Official PyTorch implementation for ICLR2025 paper "Scaling up Masked Diffusion Models on Text"

The official PyTorch implementation for Improving Long-Text Alignment for Text-to-Image Diffusion Models (LongAlign)

Official PyTorch implementation for "Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data" (ICLR 2025)

Official implementation of "MicroDreamer: Zero-shot 3D Generation in ~20 Seconds by Score-based Iterative Reconstruction".

Official PyTorch implementation for "Unifying Bayesian Flow Networks and Diffusion Models through Stochastic Differential Equations"

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

[TMLR 2025] On Memorization in Diffusion Models

Official PyTorch implementation for "Diffusion Models and Semi-Supervised Learners Benefit Mutually with Few Labels"

Official PyTorch implementation for ICLR2024 paper "The Blessing of Randomness: SDE Beats ODE in General Diffusion-based Image Editing"

[NeurIPS 2023] Official implementation for our paper "Toward Understanding Generative Data Augmentation".

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

This repository categorizes the papers about diffusion models applied in computer vision according to their target task. The classifcation is based on our survey: https://arxiv.org/abs/2209.04747v1

Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)

A temporary webpage for our survey in AGI for computer vision

Official implementation for "EGSDE: Unpaired Image-to-Image Translation via Energy-Guided Stochastic Differential Equations" (NIPS 2022)

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

[NeurIPS-2023] Annual Conference on Neural Information Processing Systems

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

Official implementation for "Revisiting Discriminative vs. Generative Classifiers: Theory and Implications".

Code for the paper Analytic-DPM: an Analytic Estimate of the Optimal Reverse Variance in Diffusion Probabilistic Models (ICLR 2022 Outstanding Paper Award)

A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".

Official implementation for Estimating the Optimal Covariance with Imperfect Mean in Diffusion Probabilistic Models (ICML 2022), and a reimplementation of Analytic-DPM: an Analytic Estimate of the …

Coupling rejection strategy against adversarial attacks (CVPR 2022)

Empirical tricks for training robust models (ICLR 2021)