jax>=0.4.20
jaxlib>=0.4.20

[benchmark]
torch>=2.0.0
torch-hd>=5.0.0
scikit-learn>=1.0.0

[datasets]
scikit-learn>=1.0.0

[dev]
pytest>=7.0
pytest-cov>=4.0
ruff>=0.4.0
mypy>=1.0

[docs]
sphinx>=4.0.0
sphinx-rtd-theme>=1.0.0
myst-parser>=0.18.0
sphinx-autodoc-typehints>=1.12.0

[examples]
matplotlib>=3.5.0
scikit-learn>=1.0.0
tqdm>=4.62.0
