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

code/api/pennylane.interfaces.jax

 

Download Python script

 

Download Notebook

 

View on GitHub