numpy
pandas
matplotlib
torch
torchvision
torchdata
tqdm
vit-pytorch
backpack-for-pytorch

