code: def test_jit_execution():
      """Test that qml.execute can be directly jitted."""
      dev = qml.device("default.qubit")
  
      tape = qml.tape.QuantumScript(
          [qml.RX(jax.numpy.array(0.1), 0)],
          [qml.expval(qml.s_prod(2.0, qml.PauliZ(0)))],
      )
  
      out = jax.jit(
          qml.execute, static_argnames=("device", "gradient_fn")
      )((tape,), device=dev, gradient_fn=qml.gradients.param_shift)
      expected = 2.0 * jax.numpy.cos(jax.numpy.array(0.1))
      assert qml.math.allclose(out[0], expected)
  
imports: import pytest
  import numpy
  from param_shift_dev import ParamShiftDerivativesDevice
  import pennylane
  from pennylane.devices import DefaultQubit
  from pennylane.gradients import param_shift
  from pennylane import execute
  from pennylane.measurements import Shots
label_desc: Use image node_id 1 for calling a global function with example usage: test_jit_execution() without return types