numpy
scikit-learn
tqdm
torch>=1.6.0

[dev]
black
isort
nbqa
flake8

[docs]
mkdocs-material

[with-hooks]
record-keeper>=0.9.32
faiss-gpu>=1.6.3
tensorboard

[with-hooks-cpu]
record-keeper>=0.9.32
faiss-cpu>=1.6.3
tensorboard
