torch
tqdm
numpy
scikit-learn

[dev]
black
pytest
pytest-cov
bump2version
pre-commit
flake8
coverage
