nnexx_ t1_irbf3qt wrote on October 6, 2022 at 7:03 PM Reply to comment by patrickkidger in [R] Introduction to Diffusion Models in JAX by jakiwjakiw Yes equinox is amazing. Also obligatory mention of diffrax that implements ODE SDE and CDE in equinox. Diffrax’s documentation even has a working continuous time diffusion model code example ! Permalink Parent 3
nnexx_ t1_irbf3qt wrote
Reply to comment by patrickkidger in [R] Introduction to Diffusion Models in JAX by jakiwjakiw
Yes equinox is amazing. Also obligatory mention of diffrax that implements ODE SDE and CDE in equinox. Diffrax’s documentation even has a working continuous time diffusion model code example !