dinosaur>=1.5.0
flax>=0.12.1
jax<0.11,>=0.10
jax-datetime>=0.1.0
tree-math
hydra-core
xarray[io,parallel]
pytest-cov
pytest-xdist
sphinx
furo
myst-parser
jax-solar
jax-rrtmgp>=0.5.0
huggingface_hub

[cosp]
jax-cosp

[era5]
gcsfs
zarr

[mam4]
mam4-jax==0.5.1

[pyses]
pyses>=0.1.3.1
