code: def test_jit_continous_gradient():
      @jit
      def f(m, op_params):
          return operators.gradient(m, params=op_params)
  
      op_params = operators.gradient.default_params(m)
      _ = f(m, op_params).params[..., 0]
  
      op_params_a = operators.gradient.default_params(a)
      _ = f(a, op_params_a).params
      _ = make_jaxpr(f)(a, op_params_a)
      _ = make_jaxpr(f)(m, op_params)
  
file_path: jaxdf/tests/test_differential.py
imports: import jax
  from jax import jit, make_jaxpr
  from jax import numpy
  from jaxdf import *
label_desc: Use image node_id 6 for calling a global function with example usage: test_jit_continous_gradient() and returns: operators