qml.interfaces.jax

This module contains functions for adding the JAX interface to a PennyLane Device class.

Functions

execute(tapes, device, execute_fn, …[, …])

Execute a batch of tapes with JAX parameters on a device.

get_jax_interface_name(tapes)

Check all parameters in each tape and output the name of the suitable JAX interface.

set_parameters_on_copy_and_unwrap(tapes, params)

Copy a set of tapes with operations and set parameters