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

[dev]
cloudpickle>=3
pytest>=8.4
pytest-cov>=7

[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
mlflow>=3
cloudpickle>=3
arviz-plots[matplotlib]>=0.6

[gpu]
jax[cuda12]>=0.7

[mlflow]
mlflow>=3
