Check all parameters in each tape and output the name of the suitable JAX interface.
This function checks each tape and determines if any of the gate parameters was transformed by a JAX transform such as
jax.jit. If so, it outputs the name of the JAX interface with jit support.
Note that determining if jit support should be turned on is done by checking if parameters are abstract. Parameters can be abstract not just for
jax.jit, but for other JAX transforms (vmap, pmap, etc.) too. The reason is that JAX doesn’t have a public API for checking whether or not the execution is within the jit transform.
tapes (Sequence[QuantumTape]) – batch of tapes to execute
name of JAX interface that fits the tape parameters, “jax” or “jax-jit”
- Return type
- What is PennyLane?
- Quantum circuits
- Gradients and training
- Quantum operators
- Inspecting circuits
- Compiling circuits
- Quantum Chemistry