diffrax<0.8,>=0.7
einops<0.9,>=0.8
equinox<0.14,>=0.13
huggingface-hub<0.36,>=0.35
jax<0.8,>=0.5
netcdf4>=1.7.2
numpy<2.4,>=2.0
xarray<2026,>=2024.1.0

[cuda]
jax[cuda]<0.8,>=0.5

[cuda12]
jax[cuda12]<0.8,>=0.5

[plots]
matplotlib>=3.10.6
seaborn>=0.13.2
cartopy>=0.25.0
regionmask>=0.13.0
pandas>=2.3.2
pyshtools>=4.13.1

[test]
pytest>=8.0.0
pytest-cov>=4.0.0
pytest-mock>=3.10.0
jaxlib>=0.4.0

[tpu]
jax[tpu]<0.8,>=0.5

[train]
dask>=2025.7.0
torch>=2.8.0
healpy>=1.18.1
optax>=0.2.5
scikit-learn>=1.7.1
scipy>=1.16.1
tqdm>=4.67.1
wandb>=0.21.3
