torch
transformers
datasets
evaluate
accelerate
scikit-learn