torch
torchaudio
torchvision
transformers
accelerate