func

Vectorise `func` over a batch axis to produce a batched function.

Build a function returning the gradient of `func`.

Return a function computing both the gradient and the primal value.

Compute the vector-Jacobian product of `func` at `primals`.

Compute the Jacobian-vector product of `func` at `primals`.

Linearise `func` around `primals` for reuse across many tangents.

Build a function returning the reverse-mode Jacobian of `func`.

Build a function returning the forward-mode Jacobian of `func`.

Build a function returning the Hessian of a scalar-valued `func`.