torch
transformers
datasets
scikit-learn
accelerate