jax
jaxlib
matplotlib
pandas
matplotlib
scipy
diffrax