metric-space/diffusion-probabilistic-models-JAX

Jax typing everything

Opened this issue 7 months ago · 1 comments

metric-space commented 7 months ago

Jax type everything to make python understandable

metric-space commented 6 months ago

Spiral code
Sohl code
Data