torch
matplotlib
transformers