einops>=0.3
filelock
numpy
torch>=1.6
