A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with simplified mathematical explanations. Made as part of my journey for learning and experimenting with state-of-the-art generative AI.
machine-learning
deep-learning
tensorflow
attention
image-generation
flax
unet
diffusion
edm
distributed-training
image2image
ai-research
jax
diffusion-models
ddpm
score-based-generative-modeling
ddim
stable-diffusion
generative-ai
karras
-
Updated
Oct 24, 2024 - Jupyter Notebook