← all repositories

NVlabs/edm

PyTorch implementation of diffusion-based generative models that achieves state-of-the-art FID scores on CIFAR-10 and ImageNet.

edm
Velocity · 7d
+1.5
★ / day
Trend
steady
star history

This repository implements a research paper that analyzes and clarifies the design space of diffusion-based generative models. It provides training and sampling code for score networks that generate images via stochastic denoising processes. The work introduces improved preconditioning, sampling schedules, and training methodologies that together achieve FID scores of 1.79 on CIFAR-10 (class-conditional) and 1.97 (unconditional), with faster convergence than prior approaches.

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