jax
jaxlib
diffrax
numpy
matplotlib
scipy
pandas
