torch
transformers
scikit-learn
joblib
numpy
accelerate