transformers
torch
accelerate
scikit-learn