David Vigouroux

When using the DormandPrince solver for ode, the second derivative is always nan. Is it a bug or is it not supported? The BDF solver supports the second derivative. tensorflow...

Memory leak for matmul operation

4

Hi, In my use case, I run several matmul operations and I observe that the memory used by my program grows very fast. I try to investigate why my memory...

There is a strange behavoir when using seal_poly_eval with the 3th parameter of coeffs different than 0 (in the unitary test the value is 0): ```python with tf.Session() as sess:...

SaveAt dense and fn

1

When using dense=True in SaveAt and a custom fn, I was expected that the interpolation of the sol will respect the custom fn but it's not the case. For now,...

Pretrained models are not accesible

4

When I try to access to the page of pretrain model (http://sceneparsing.csail.mit.edu/model/pytorch/), the following error is raised: You don't have permission to access this resource. Could you provide access to...

Event and PIDController: event doesn't always occure

3

First, I want to thank you for your amazing library. You have done a massive work which are very useful for my research. ----------------- diffrax 0.6.0 optimistix 0.0.7 jax 0.4.30...

jax.experimental.enable_x64 and jit

1

diffrax 0.6.0 jax 0.4.30 Using jax.experimental.enable_x64 and jit will return a exception. ```python import jax import jax.numpy as jnp import diffrax key = jax.random.PRNGKey(0) x1 = jax.random.uniform(key,(2,)) x2 = jax.random.uniform(key,(2,2))...