numpy
scipy
matplotlib
jax
jaxlib
diffrax
equinox
