torch
torchvision
numpy
pandas
umap-learn
matplotlib
scikit-learn
argparse
