torch
torchvision
itertools
matplotlib
numpy
sklearn
pandas
umap
