numpy>=1.19.0
einops>=0.3.0
torch>=1.6.0
