torch.func.jvp
torch.func.jvp(func, primals, tangents, *, strict=False, has_aux=False)
Standing for the Jacobian-vector product, returns a tuple
   containing the output of func(primals)* and the "Jacobian of
   "func" evaluated at "primals"" times "tangents". This is also known
   as forward-mode autodiff.
Parameters:
      * func (function) -- A Python function that takes one or
        more arguments, one of which must be a Tensor, and returns one
        or more Tensors
  * **primals** (*Tensors*) -- Positional arguments to "func" that
    must all be Tensors. The returned function will also be
    computing the derivative with respect to these arguments

  * **tangents** (*Tensors*) -- The "vector" for which Jacobian-
    vector-product is computed. Must be the same structure and
    sizes as the inputs to "func".

  * **has_aux** (*bool*) -- Flag indicating that "func" returns a
