jax

ba29d5a0 - Merge pull request #23821 from jakevdp:jnp-doc-examples