streamlit numpy matplotlib jaxlib jax tensorflow tensorflow_probability seaborn