numpy
torchvision
timm