Diffusion model research harness built on Keras 3 + JAX — unconditional DDPM, classifier-free guidance, and more - View it on GitHub