jax
jaxlib
matplotlib
diffrax
scipy
netcdf4