Adam Hartshorne
Adam Hartshorne
In function curvature.m, the return value 'l' is never set.
In this new paper, the authors argue that the diffusion parameter (output scale in probdiff) should be set on a decreasing schedule (rather than learned in the case of Fenrir)....
The newly introduced requirement (as part of the code refactorisation) of having to state the implementation AND the ode_shape a-priori can be slightly problematic. If one is using an ode_solve...
I have run the following MVE versus Distrax (https://github.com/google-deepmind/distrax) and your library doesn't seem to be as fast. I am running this using jax 0.4.23, cuda 12.2, python 3.10 on...