qml.workflow.interfaces.jax_jit.jax_jit_vjp_execute

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

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