plotly torch scipy scikit-learn einops