numpy
scipy
scikit-learn
torch
tqdm
umap-learn
pynndescent

[dev]
matplotlib
pytest

[gpu]
faiss-gpu
