torch
umap-learn
plotly
pandas
transformers