numpy
scipy
plotly
torch
torchvision
tqdm
