torch
transformers
numpy
scikit-learn
datasets
accelerate