qml.workflow.interfaces.jax.jax_jvp_execute

jax_jvp_execute(tapes, execute_fn, jpc, device=None)

Execute a batch of tapes with JAX parameters using JVP derivatives.

Parameters

• tapes (Sequence[QuantumTape]) – batch of tapes to execute

• execute_fn (Callable[[Sequence[QuantumTape]], ResultBatch]) – a function that turns a batch of circuits into results

• jpc (JacobianProductCalculator) – a class that can compute the Jacobian vector product (JVP) for the input tapes.

Returns

A nested tuple of tape results. Each element in the returned tuple corresponds in order to the provided tapes.

Return type

TensorLike

code/api/pennylane.workflow.interfaces.jax.jax_jvp_execute

 

Download Python script

 

Download Notebook

 

View on GitHub