torch
torchvision
numpy
scipy
matplotlib
umap-learn
scikit-learn
tqdm
