dask>=2025.10
jax<0.10,>=0.9
numpyro>=0.20.0
scikit-learn>=1.6.1
xarray>=2025.10
zarr<4,>=3

[dev]
pre-commit>=4.5
pytest>=9
pytest-cov>=7
ruff>=0.15

[docs]
flax>=0.12
jupyter-sphinx>=0.5
mlflow>=3.5
myst-parser>=5
pydata-sphinx-theme>=0.16
sphinx>=9
sphinx-copybutton>=0.5
sphinx-design>=0.7

[docs:python_version >= "3.12"]
arviz_base>=1
arviz_plots>=1
arviz_stats>=1

[gpu]
jax[cuda13]>=0.9

[mlflow]
mlflow>=3.5
