numpy
matplotlib
torch
ipython
tqdm
