torch>=1.6.0a
numpy
timm
einops
