scipy
numpy
torch
tqdm