torch
numpy
tqdm
matplotlib