jax

f37f2351 - Fix up previous jaxpr.rst commit. (#2647)