jax.value\_and\_grad ==================== .. currentmodule:: jax .. autofunction:: value_and_grad