flaxdiff
Jupyter Notebook
A simple, easy-to-understand library for diffusion models built with Flax and JAX, focusing on readability and learning.
41 stars1 forks0 contributors
Overview
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.
Related Projects
Community-curated · Updated weekly · 100% open source
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
