/tiny-diffusion

A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.

Primary LanguageJupyter Notebook

Stargazers