torch
transformers
matplotlib