code: def map(f, xs):  # pylint: disable=redefined-builtin
      """Equivalent to :func:`jax.lax.map` but with Haiku state passed in/out."""
      g = lambda _, x: ((), f(x))
      _, ys = scan(g, (), xs)
      return ys
  
imports: import collections
  import collections.abc
  from collections.abc import Mapping, MutableMapping
  import functools
  import inspect
  from typing import Any, Callable, Optional, TypeVar
  from haiku._src import base
  import jax
  import jax.numpy
label_desc: Use image node_id 20 for calling a global function with example usage: map(f, xs) and returns: ys