philippe-eecs/JaxDDPM

A DDPM implementation in Jax for continuous space modeling.

PythonMIT