numpy
scipy
torch
faiss-cpu
tqdm
