GitHub repository: tux-type/simple-diffusion
Primary language: Python
Description: Simple diffusion in Flax NNX.