einops>=0.6
filelock
joblib
faiss-gpu
numpy
torch>=1.6
