numpy
torchvision
timm>=0.5.4