dask>=2025.7
jax>=0.8
numpyro>=0.19.0
scikit-learn>=1.6.1
xarray>=2025.7
zarr<4,>=3

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

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

[gpu]
jax[cuda13]>=0.7

[mlflow]
mlflow>=3
