jax
jaxlib
matplotlib
diffrax
scipy
netcdf4
f90nml
pyevtk