dask>=2025.5
jax>=0.7
numpyro>=0.19.0
scikit-learn>=1.6.1
xarray>=2025.4
zarr<4,>=3

[dev]
pre-commit>=4.3
pytest>=8.4
pytest-cov>=7
ruff>=0.14

[docs]
sphinx>=8.2
myst-parser>=4.0
jupyter-sphinx>=0.5
sphinx-copybutton>=0.5
sphinx-design>=0.6
pydata-sphinx-theme>=0.16
flax>=0.12
mlflow>=3
arviz-plots[matplotlib]>=0.6

[gpu]
jax[cuda13]>=0.7

[mlflow]
mlflow>=3
