andylolu2/jax-diffusion

Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.

Python