pandas jax matplotlib