qml.workflow.interfaces.jax_jit.jax_jit_jvp_execute

jax_jit_jvp_execute(tapes, execute_fn, jpc, device)

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 for the input tapes.

• device (pennylane.Device, pennylane.devices.Device) – The device used for execution. Used to determine the shapes of outputs for pure callback calls.

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_jit.jax_jit_jvp_execute

 

Download Python script

 

Download Notebook

 

View on GitHub