numpy
pandas
torch
torchvision
transformers
accelerate
datasets