dask>=2026.1
jax>=0.10
numpyro>=0.22
psutil>=6
xarray>=2026.1
zarr<4,>=3

[dev]
mlflow>=3.10
pre-commit>=4.6
pytest>=9.1
pytest-cov>=7.1
ruff>=0.16

[docs]
arviz_base>=1
arviz_plots>=1
arviz_stats>=1
flax>=0.12
jupyter-sphinx>=0.5
mlflow>=3.10
myst-parser>=5
pydata-sphinx-theme>=0.22
sphinx>=9.1
sphinx-copybutton>=0.5
sphinx-design>=0.7

[gpu]
jax[cuda13]>=0.10

[mlflow]
mlflow>=3.10
