docstr: Test with JAX interface
code: def test_get_unitary_matrix_interface_jax(self):
      """Test with JAX interface"""
  
      from jax import numpy as jnp
  
      @qml.matrix
      def circuit(theta):
          qml.RZ(theta[0], wires=0)
          qml.RZ(theta[1], wires=1)
          qml.CRY(theta[2], wires=[1, 2])
  
      # input jax parameters
      theta = jnp.array([0.1, 0.2, 0.3], dtype=jnp.float64)
  
      matrix = circuit(theta)
  
      # expected matrix
      matrix1 = np.kron(
          qml.RZ(theta[0], wires=0).matrix(),
          np.kron(qml.RZ(theta[1], wires=1).matrix(), I),
      )
      matrix2 = np.kron(I, qml.CRY(theta[2], wires=[1, 2]).matrix())
      expected_matrix = matrix2 @ matrix1
  
      assert np.allclose(matrix, expected_matrix)
  
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 4 for calling the TestInterfaces obj's underlying member method code with example usage: obj.test_get_unitary_matrix_interface_jax() without return types