code: def scalar_mul(x, y):
      assert np.shape(x) == np.shape(y) == ()
      return x * y
  
imports: from functools import partial
  from absl.testing import absltest
  import numpy
  import jax
  import jax.numpy
  from jax._src import test_util
  from jax._src.numpy.ufunc_api import get_if_single_primitive
  from jax import config
label_desc: Use image node_id 3 for calling a global function with example usage: scalar_mul(x, y) and returns: unknown