jax

bbf625e0 - fix jax.rst docs (remove defvjp2 / defjvp2)