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