GitHub / AshishKumar4 / FlaxDiff
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.
JSON API: http://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/AshishKumar4%2FFlaxDiff
PURL: pkg:github/AshishKumar4/FlaxDiff
Stars: 25
Forks: 0
Open issues: 0
License: None
Language: Jupyter Notebook
Size: 238 MB
Dependencies parsed at: Pending
Created at: about 1 year ago
Updated at: 3 months ago
Pushed at: 3 months ago
Last synced at: 3 months ago
Topics: ai-research, attention, ddim, ddpm, deep-learning, diffusion, diffusion-models, distributed-training, edm, flax, generative-ai, image-generation, image2image, jax, karras, machine-learning, score-based-generative-modeling, stable-diffusion, tensorflow, unet