jvp#

bayesflow.utils.jvp(f: Callable, x: Tensor | tuple[Tensor, ...], tangents: Tensor | tuple[Tensor, ...], return_output: bool = False)[source]#

Compute the Jacobian-vector product of f at x with tangents.