torch==1.9.0
timm==0.3.4
einops
ml_collections
