JVP

Compute the Jacobian-vector product.

Abstract Signature:

JVP(func: Callable, primals: List[Tensor], tangents: List[Tensor])

PyTorch

API: torch.autograd.functional.jvp
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.jvp
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.linalg.jvp
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.jvp
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: nnx.variablelib.GetVariable.jvp
Strategy: Direct Mapping
Official Docs ↗