scikit-learn
numpy
transformers
accelerate
torch
