andylolu2 / jax-diffusion

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

Geek Repo

Github PK Tool