numpy
torch
transformers
accelerate
matplotlib
seaborn