Simple Diffusion

A simple diffusion implementation in Flax NNX, based on this amazing repo: https://github.com/cloneofsimo/minDiffusion.

Samples

MNIST 10 Epochs

Flowers 85 Epochs

Flowers 200 Epochs

Name		Name	Last commit message	Last commit date
Latest commit History 17 Commits
diffusion		diffusion
imgs		imgs
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
pyproject.toml		pyproject.toml
train_flowers.py		train_flowers.py
train_mnist.py		train_mnist.py