jax

604cdbcf - Merge pull request #20323 from jakevdp:fix-doctest