Francois Lanusse

Computing growth is tiny tiny tiny compared to the computational budget of a reasonable sim, no?

I would personally opt to use proper solvers instead of emulators wherever you can afford it. But that's my preference 😉

Oh right right, well the problem is that ultimately there will need to be a jit outside of pmwd. For instance in an hmc sampler or as part a of...

When you use an external sampler like numpyro or TFP, it will usually compile all the logic of the hmc kernel, including the evaluation of the log likelihood. It may...

And actually ^^ it's generally not a good idea ^^' but if we want to, we can definitely write a custom CPU op that will dump the simulation in hdf5...

So yeah I don't see any drawbacks of using a scan :-)

In odeint they use a while inside the scan function yes. Would you be ok with an API with an argument which would be the array `a` to use in...

diffrax is nice for sure, but I don't think it will make a big difference (I may be wrong), It does give you more control over the ODE solver, and...

If someone could look at the demo notebook and explain to me why the two histograms don't match, I would be sooooo happy.

Thanks so much!