Viewing a single comment thread. View all comments

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 !