numpy
pandas
torch
torchvision
transformers
accelerate
datasets
os
PIL
itertools