torch
Pillow
numpy
matplotlib
timm==0.4.5