qml.interfaces.jax_jit

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

Functions

execute_legacy(tapes, device, execute_fn, …)

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

code/api/pennylane.interfaces.jax_jit

 

Download Python script

 

Download Notebook

 

View on GitHub