matplotlib
torch
transformers
datasets
num2words
accelerate