scipy
torch
torchvision
transformers
accelerate