class_docstr: Tests Von Neumann entropy transform
docstr: Test entropy for a QNode gradient with Jax-jit.
docstr_tok: ["Test","entropy","for","a","QNode","gradient","with","Jax-jit","."]
code: def test_IsingXX_qnode_entropy_grad_jax_jit(
      self, param, wires, base, interface
  ):
      """Test entropy for a QNode gradient with Jax-jit."""
      import jax
  
      dev = qml.device("default.qubit", wires=2)
  
      @qml.qnode(dev, interface=interface, diff_method="backprop")
      def circuit_state(x):
          qml.IsingXX(x, wires=[0, 1])
          return qml.state()
  
      grad_entropy = jax.jit(
          jax.grad(
              qml.qinfo.vn_entropy(
                  circuit_state, wires=wires, base=base
              )
          )
      )(jax.numpy.array(param))
  
      grad_expected_entropy = expected_entropy_grad_ising_xx(
          param
      ) / np.log(base)
  
      assert qml.math.allclose(
          grad_entropy, grad_expected_entropy, rtol=1e-04, atol=1e-05
      )
  
imports: import pytest
  import pennylane
  from pennylane import numpy
label_desc: Use image node_id 11 for calling the TestVonNeumannEntropy obj's underlying member method code with example usage: obj.test_IsingXX_qnode_entropy_grad_jax_jit(param, wires, base, interface) and returns: qml