class_docstr: Test that rotation merging works in all interfaces.
docstr: Test that when using jax.jit, the conditional statement that checks for
  0 rotation angles does not break things.
docstr_tok: ["Test","that","when","using","jax.jit",",","the","conditional","statement","that","checks","for","0","rotation","angles","does","not","break","things","."]
code: def test_merge_rotations_jax_jit(self):
      """Test that when using jax.jit, the conditional statement that checks for
      0 rotation angles does not break things."""
  
      import jax
  
      @jax.jit
      @qml.qnode(
          qml.device("default.qubit", wires=["w1", "w2"]),
          interface="jax",
      )
      @merge_rotations
      def qfunc():
          qml.Rot(
              jax.numpy.array(0.1),
              jax.numpy.array(0.2),
              jax.numpy.array(0.3),
              wires=["w1"],
          )
          qml.Rot(
              jax.numpy.array(-0.1),
              jax.numpy.array(-0.2),
              jax.numpy.array(-0.3),
              wires=["w1"],
          )
          qml.CRX(jax.numpy.array(0.2), wires=["w1", "w2"])
          qml.CRX(jax.numpy.array(-0.2), wires=["w1", "w2"])
          return qml.expval(qml.PauliZ("w1"))
  
      res = qfunc()
  
      assert qml.math.allclose(res, [1.0])
  
imports: import pytest
  from utils import compare_operation_lists
  import pennylane
  from pennylane import numpy
  from pennylane.wires import Wires
  from pennylane.transforms.optimization import merge_rotations
label_desc: Use image node_id 5 for calling the TestMergeRotationsInterfaces obj's underlying member method code with example usage: obj.test_merge_rotations_jax_jit() and returns: qml