ECLIPSE-AI4Science / gradsolve Differentiable ensemble solvers for differential equations on GPUs, in JAX