numpy
torch
torchvision
timm
tensorboard
