yang-song/score_sde
Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
Unifies score-based generative models through continuous-time SDEs, enabling unified training of NCSN, NCSNv2, DDPM variants, and new NCSN++/DDPM++ architectures. Supports exact likelihood computation, latent code manipulation, and conditional generation (inpainting, colorization, class-conditional) through flexible Predictor-Corrector sampling. Implemented in JAX with modular abstractions for SDEs, predictors, and correctors, allowing straightforward extensions while maintaining compatibility across sampling and evaluation methods.
1,811 stars. No commits in the last 6 months.
Stars
1,811
Forks
230
Language
Jupyter Notebook
License
Apache-2.0
Category
Last pushed
Nov 29, 2022
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/diffusion/yang-song/score_sde"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Compare
Higher-rated alternatives
yang-song/score_sde_pytorch
PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential...
homerjed/sbgm
Score-based Diffusion models in JAX.
ermongroup/ncsnv2
The official PyTorch implementation for NCSNv2 (NeurIPS 2020)
aurelio-amerio/GenSBI
Generative Models for Simulation-Based Inference in JAX
amazon-science/unconditional-time-series-diffusion
Official PyTorch implementation of TSDiff models presented in the NeurIPS 2023 paper "Predict,...