torch
numpy
transformers
matplotlib
