transformers
datasets
torch
scikit-learn
tensorboard