jax

cdc72787 - Merge pull request #24025 from jakevdp:gradient-doc