numpy
matplotlib
tqdm
torch
