← all repositories

yang-song/score_sde

JAX/Flax implementation of score-based diffusion models via stochastic differential equations for high-fidelity image generation.

1.8k stars Jupyter Notebook Image · Video · AudioML Frameworks
score_sde
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

This repository provides the official implementation of score-based generative modeling through stochastic differential equations, a unified framework that generalizes diffusion models. The method transforms data to noise via continuous-time stochastic processes and reverses the process for generation using learned score functions. It supports unconditional generation, class-conditional generation, inpainting, colorization, and latent code manipulation, achieving FID of 2.20 on CIFAR-10 and high-fidelity 1024px CelebA-HQ generation.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.