code: def scalar_mul(xs, a):
      def mul(x):
          dtype = _dtype(x)
          result = np.multiply(x, np.array(a, dtype=dtype), dtype=dtype)
          return result.item() if is_python_scalar(x) else result
  
      return tree_map(mul, xs)
  
imports: from functools import partial
  import operator
  from jax._src import api
  from jax._src import config
  from jax._src import dtypes
  from jax._src.tree_util import tree_map, tree_reduce
  import numpy
label_desc: Use image node_id 12 for calling a global function with example usage: scalar_mul(xs, a) and returns: tree_map, unknown