Score-based Diffusion models in JAX.
Combinatorial Complex Score-based Diffusion model using stochastic differential equations
