numpy
scipy
matplotlib
diffrax
equinox
interpax
jax
pytest
