torch numpy matplotlib scikit-learn umap-learn plotly pandas