qml.workflow.interfaces.jax_jit.jax_jit_vjp_execute¶
- jax_jit_vjp_execute(tapes, execute_fn, jpc, device=None)[source]¶
Execute a batch of tapes with JAX parameters using VJP 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 vector Jacobian product (VJP) 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_vjp_execute
Download Python script
Download Notebook
View on GitHub