Awesome

JaxDDPM

A DDPM implementation in Jax for continuous space modeling. Complementary to https://github.com/philippe-eecs/IDQL.

Credit to Ilya Kostrikov for help in making this repo!