einops>=0.4
einops-exts
filelock
joblib
faiss-gpu
numpy
torch>=1.6
