torch>=1.6.0a
numpy
timm<1.0.23
einops
