Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

Official code for "Maximum Likelihood Training for Score-Based Diffusion ODEs by High-Order Denoising Score Matching" (ICML 2022)

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

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

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.

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

code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"

Karras et al. (2022) diffusion models for PyTorch

Implementation of Alphafold 3 from Google Deepmind in Pytorch

Implementation of Alpha Fold 3 from the paper: "Accurate structure prediction of biomolecular interactions with AlphaFold3" in PyTorch

A summary of related works about flow matching, stochastic interpolants