Interest in folding in diffraxtra

[nstarman]

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

[patrick-kidger]

Haha, I love the pun in the name!

On balance I'm inclined to leave these in diffraxtra. I think utilities like these are a great example of how people can customise wrappers to fit their needs, but I'd like to keep the underlying library doing each thing in 'just one way'.

Thank you for the offer though! :)