jax

dbda9168 - Merge pull request #38295 from jakevdp:doc-jit-partial