transformers
datasets
torch
numpy
accelerate