numpy
scikit-learn
torch>=1.6.0
tqdm

[dev]
black
flake8
isort
nbqa

[docs]
mkdocs-material

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

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