torch
transformers
tqdm
matplotlib