numpy
scipy
matplotlib
torch==1.9.0
tqdm
termcolor
