Harrison Zhu

Hi all, thank you for the great work! I was just trying to run the code from the pull request and encountered this error ``` NotImplementedError Traceback (most recent call...

> Hmm. This looks like a bug in core JAX -- `closed_call` looks like a new JAX primitive, which `jax.experimental.host_callback.call` hasn't been updated to be able to handle. > >...

> Hi @harrisonzhu508, you can run the current code with `jax=0.3.15` (while waiting for further update). The latest version of `equinox=0.7.1` works fine. > > As Patrick mentioned, it must...

Hi @anh-tong, thanks a lot for the very clean implementation again! I was trying to reproduce an example that is very similar to https://github.com/google-research/torchsde/blob/master/examples/latent_sde.py. Running the latter script yields the...

> > Hi @anh-tong, thanks a lot for the very clean implementation again! I was trying to reproduce an example that is very similar to https://github.com/google-research/torchsde/blob/master/examples/latent_sde.py. Running the latter script...