dask>=2026.1
jax>=0.10
numpyro>=0.21
xarray>=2026.1
zarr<4,>=3

[dev]
mlflow>=3.9
pre-commit>=4.6
pytest>=9.1
pytest-cov>=7.1
ruff>=0.15

[docs]
flax>=0.12
jupyter-sphinx>=0.5
mlflow>=3.9
myst-parser>=5
pydata-sphinx-theme>=0.17
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.10

[mlflow]
mlflow>=3.9
