jax.vjp ======= .. currentmodule:: jax .. autofunction:: vjp