torch
torchvision
matplotlib
numpy
sklearn
pandas
umap-learn
