[BUG] cancel_inverses is not jittable

[albi3ro]

Expected behavior

I expect cancel_inverses to be jittable.

Actual behavior

We get an error because we are trying to compare the equality of tracers.

Additional information

No response

Source code

@qml.transforms.cancel_inverses
@qml.qnode(qml.device('default.qubit'))
def circuit(x):
    qml.adjoint(qml.RX(x+0.1, 0))
    qml.RX(x,0)
    return qml.expval(qml.Z(0))

jax.jit(circuit)(jax.numpy.array(0.5))

Tracebacks

TracerBoolConversionError: Attempted boolean conversion of traced array with shape bool[]..
The error occurred while tracing the function circuit at /var/folders/k1/0v_kvphn55lgf_45kntf1hqm0000gq/T/ipykernel_91848/3926972727.py:1 for jit. This concrete value was not available in Python because it depends on the value of the argument x.
See https://jax.readthedocs.io/en/latest/errors.html#jax.errors.TracerBoolConversionError

System information

master

Existing GitHub issues

• I have searched existing GitHub issues to make sure the issue does not already exist.