torch
accelerate>=0.20.1
transformers
scikit-learn
pandas
numpy