numpy
wandb
faiss-cpu
scikit-learn
torch
torchvision
torchaudio
