numpy<2
scikit-learn
torch

[dev]
pytest
mkdocs-material
mkdocstrings
ipywidgets
