README.md

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