torch>=1.11
scikit-learn
matplotlib
sortedcontainers
tqdm

[dev]
pre-commit
sphinx
black
