torch>=2.0
numpy>=1.21

[dev]
pytest>=7
pytest-cov>=4
ruff>=0.4
build>=1.0
twine>=5.0

[docs]
mkdocs>=1.5
mkdocs-material>=9.0
mkdocstrings[python]>=0.24

[sklearn]
scikit-learn>=1.0
