https://github.com/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.
https://github.com/AshishKumar4/FlaxDiff

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

Last synced: 30 days ago
JSON representation

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.

• Host: GitHub
• URL: https://github.com/AshishKumar4/FlaxDiff
• Owner: AshishKumar4
• Created: 2024-06-17T11:53:58.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2025-05-02T00:15:25.000Z (5 months ago)
• Last Synced: 2025-05-02T01:23:36.034Z (5 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
• Language: Jupyter Notebook
• Homepage:
• Size: 238 MB
• Stars: 25
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• awesome-jax - flaxdiff - Framework and Library for building and training Diffusion models in multi-node multi-device distributed settings (TPUs) <img src="https://img.shields.io/github/stars/AshishKumar4/FlaxDiff?style=social" align="center"> (Libraries / New Libraries)