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.

Simple KES blends Karras and Exponential noise scheduling for high-quality images with adaptable detail, real-time config reloading, and parameter randomization for creative flexibility.