qml.interfaces.jax_jit

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.