torch
transformers
matplotlib
sklearn
numpy