transformers
datasets
scikit-learn
torch