Hi @patrick-kidger!
I recently moved some utilities for working with diffrax out of galax and into its own library — diffraxtra. I was wondering if you'd be interested in folding any of these into diffrax proper.

Currently in diffraxtra:

• DiffEqSolver: an object-oriented interface to diffrax.diffeqsolve.
• VectorizedDenseInterpolation: a vectorized form of diffrax.DenseInterpolation that works on batched results from diffrax.diffeqsolve