numpy
transformers
torch
accelerate
nltk