class_docstr: Test vector-valued returns for the JAX-JIT interface.
docstr: Test the jacobian computation with multiple tapes with probability
  and expectation value computations.
docstr_tok: ["Test","the","jacobian","computation","with","multiple","tapes","with","probability","and","expectation","value","computations","."]
code: def test_multi_tape_jacobian_probs_expvals(self, execute_kwargs):
      """Test the jacobian computation with multiple tapes with probability
      and expectation value computations."""
      adjoint = (
          execute_kwargs.get("gradient_kwargs", {}).get("method", "")
          == "adjoint_jacobian"
      )
      if adjoint:
          pytest.skip(
              "The adjoint diff method doesn't support probabilities."
          )
  
      def cost(x, y, device, interface, ek):
          with qml.queuing.AnnotatedQueue() as q1:
              qml.RX(x, wires=[0])
              qml.RY(y, wires=[1])
              qml.CNOT(wires=[0, 1])
              qml.expval(qml.PauliZ(0))
              qml.expval(qml.PauliZ(1))
  
          tape1 = qml.tape.QuantumScript.from_queue(q1)
  
          with qml.queuing.AnnotatedQueue() as q2:
              qml.RX(x, wires=[0])
              qml.RY(y, wires=[1])
              qml.CNOT(wires=[0, 1])
              qml.probs(wires=[0])
              qml.probs(wires=[1])
  
          tape2 = qml.tape.QuantumScript.from_queue(q2)
  
          return qml.execute(
              [tape1, tape2], device, **ek, interface=interface
          )[0]
  
      dev = qml.device("default.qubit.legacy", wires=2)
      x = jax.numpy.array(0.543)
      y = jax.numpy.array(-0.654)
  
      x_ = np.array(0.543)
      y_ = np.array(-0.654)
  
      res = cost(x, y, dev, interface="jax-jit", ek=execute_kwargs)
  
      exp = cost(x_, y_, dev, interface="autograd", ek=execute_kwargs)
  
      for r, e in zip(res, exp):
          assert jax.numpy.allclose(r, e, atol=1e-7)
  
imports: import pytest
  import numpy
  import pennylane
  from pennylane.gradients import param_shift
  from pennylane import execute
label_desc: Use image node_id 6 for calling the TestVectorValuedJIT obj's underlying member method code with example usage: obj.test_multi_tape_jacobian_probs_expvals(execute_kwargs) and returns: qml