netCDF4
numpy
pyyaml
tqdm
xarray
jax
jaxlib
jaxtyping
optax

[dev]
pytest
sphinx
myst_parser
twine
build

[notebook]
jupyterlab
matplotlib
