numpy
torch
datasets
transformers
argparse
matplotlib
scipy
tensorboard
tensorflow