jax

8337fe5b - Merge pull request #29563 from jakevdp:fix-doc