class_docstr: Autodiff tests for all frameworks
class_docstr_tok: ["Autodiff","tests","for","all","frameworks"]
docstr: Test that results after splitting are still differentiable with jax-jit
docstr_tok: ["Test","that","results","after","splitting","are","still","differentiable","with","jax-jit"]
code: def test_split_with_jax_jit(self):
      """Test that results after splitting are still differentiable with jax-jit"""
  
      import jax
      import jax.numpy as jnp
  
      dev = qml.device("default.qubit", wires=3)
  
      @jax.jit
      @qml.qnode(dev)
      def circuit(params):
          qml.RX(params[0], wires=0)
          qml.RY(params[1], wires=1)
          return (
              qml.expval(qml.PauliZ(0) @ qml.PauliZ(1)),
              qml.expval(qml.PauliY(0)),
              qml.expval(qml.PauliZ(0)),
          )
  
      params = jnp.array([0.5, 0.5])
      res = circuit(params)
      grad = jax.jacobian(circuit)(params)
      assert all(np.isclose(res, exp_res))
      assert all(np.isclose(grad, exp_grad, atol=1e-5).flatten())
  
imports: import itertools
  import pytest
  import numpy
  import pennylane
  import pennylane.numpy
  from pennylane.transforms import split_non_commuting
label_desc: Use image node_id 7 for calling the TestAutodiffSplitNonCommuting obj's underlying member method code with example usage: obj.test_split_with_jax_jit() and returns: qml,  qml,  qml