code: def test_jitting_matrix():
      """Test that qml.matrix is jittable with jax."""
      import jax
  
      op = qml.adjoint(qml.Rot(1.2, 2.3, 3.4, wires=0))
  
      jit_mat = jax.jit(qml.matrix)(op)
      normal_mat = qml.matrix(op)
  
      assert qml.math.allclose(normal_mat, jit_mat)
  
imports: from functools import reduce, partial
  from warnings import catch_warnings
  import pytest
  from gate_data import I, X, Y, Z, H, S, CNOT, Rotx, Roty
  import pennylane
  from pennylane import numpy
  from pennylane.transforms.op_transforms import OperationTransformError
label_desc: Use image node_id 1 for calling a global function with example usage: test_jitting_matrix() without return types