torch
numpy
matplotlib
tqdm
scipy
