jax

ce51384b - Merge pull request #26375 from jakevdp:fix-doc