Start update to jax 3.1.0