Closed bnishanth16 closed 3 weeks ago
Hi @bnishanth16, thank you so much for reporting this! I'll forward it to the PennyLane team since this is an issue with the qml.exp
function not being traceable (that is the same error occurs when jax.jit
is used).
Experienced error when trying to parameterize qml.exp with qjit. Example code below.
Expected behavior:
Output:
Actual behavior: with qjit
output:
import pennylane as qml; qml.about()
)Additional information: replacing
qml.exp(qml.X(0), coeff=-1j*jnp.cos(theta))
withqml.evolve(qml.X(0), coeff = jnp.cos(theta))
causes the same errorTracerBoolConversionError
, but replacing it withqml.RX(2*jnp.cos(theta),0)
doesn't cause the error to occur.