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