code: def rosenbrock(np):
      def func(x):
          return np.sum(100.0 * np.diff(x) ** 2 + (1.0 - x[:-1]) ** 2)
  
      return func
  
imports: from absl.testing import absltest
  import numpy
  import scipy
  import scipy.optimize
  from jax import numpy
  from jax._src import test_util
  from jax import jit
  from jax import config
  import jax.scipy.optimize
label_desc: Use image node_id 1 for calling a global function with example usage: rosenbrock(np) and returns: func, np