GitHub repository: AshishKumar4/FlaxDiff
Primary language: Jupyter Notebook
Created: 2024-06-17T11:53:59Z
Description: 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 generative AI.