code: def test_diff_method_None_jit():
      """Test that jitted execution works when `gradient_fn=None`."""
  
      dev = qml.device("default.qubit.jax", wires=1, shots=10)
  
      @jax.jit
      def wrapper(x):
          with qml.queuing.AnnotatedQueue() as q:
              qml.RX(x, wires=0)
              qml.expval(qml.PauliZ(0))
  
          tape = qml.tape.QuantumScript.from_queue(q)
  
          return qml.execute([tape], dev, gradient_fn=None)
  
      assert jax.numpy.allclose(wrapper(jax.numpy.array(0.0))[0], 1.0)
  
imports: import pytest
  import numpy
  import pennylane
  from pennylane.gradients import param_shift
  from pennylane import execute
label_desc: Use image node_id 1 for calling a global function with example usage: test_diff_method_None_jit() and returns: qml