numpy
torch
pytorch-lightning
matplotlib
scikit-learn
pandas
umap-learn