numpy
scipy
matplotlib
diffrax
equinox==0.13.2
interpax
jax==0.8.1
pytest
optimistix==0.0.11
