jax

fc654a57 - Merge pull request #20732 from jakevdp:doc-stateful