sentence_transformers pandas jax jaxlib streamlit numpy torch scikit-learn plotly matplotlib