scikit-learn
torch

[dev]
pytest
mkdocs-material
mkdocstrings
