matplotlib
torch
transformers