class_docstr: Tests for the (reduced) density matrix for QNodes returning states.
docstr: Test jitting the density matrix from state vector function with Tf.
docstr_tok: ["Test","jitting","the","density","matrix","from","state","vector","function","with","Tf","."]
code: def test_density_matrix_qnode_tf_jit(self):
      """Test jitting the density matrix from state vector function with Tf."""
      dev = qml.device("default.qubit", wires=2)
  
      @qml.qnode(dev, interface="tf")
      def circuit(x):
          qml.IsingXX(x, wires=[0, 1])
          return qml.state()
  
      density_matrix = tf.function(
          qml.qinfo.reduced_dm(circuit, wires=[0]),
          jit_compile=True,
          input_signature=(tf.TensorSpec(shape=(), dtype=tf.float32),),
      )
      density_matrix = density_matrix(
          tf.Variable(0.0, dtype=tf.float32)
      )
      assert np.allclose(density_matrix, [[1, 0], [0, 0]])
  
file_path: pennylane/tests/qinfo/test_reduced_dm.py
imports: import pytest
  import pennylane
  from pennylane import numpy
label_desc: Use image node_id 6 for calling the TestDensityMatrixQNode obj's underlying member method code with example usage: obj.test_density_matrix_qnode_tf_jit() and returns: qml