torch
tqdm
matplotlib
