jax>=0.6.2
jaxlib>=0.6.2
equinox>=0.13.3
jaxtyping>=0.2.24
tqdm
matplotlib
diffrax>=0.7.0
optax
jaxopt
optimistix
scipy
jaxkd
netcdf4
pyevtk
