jax
jaxlib
tqdm
matplotlib
diffrax
optax
jaxopt
optimistix
scipy
jaxkd
netcdf4
pyevtk
