code: def _pallas_call_abstract_eval(*avals, out_shapes, **_):
      return map(
          lambda x: jax_core.ShapedArray(x.shape, x.dtype), out_shapes
      )
  
imports: from __future__ import annotations
  from functools import partial
  import itertools
  from typing import Any, Callable
  from collections.abc import Sequence
  import jax
  from jax import api_util
  from jax import tree_util
  from jax import lax
  from jax._src import state
  from jax.interpreters import ad
  from jax.interpreters import batching
  from jax.interpreters import partial_eval
  from jax.interpreters import mlir
  from jax.interpreters import xla
  from jax._src import ad_util
  from jax._src import core
  from jax._src.state import primitives
  from jax._src import linear_util
  from jax._src.state import discharge
  from jax._src.util import split_list, safe_map, safe_zip, weakref_lru_cache, tuple_insert, partition_list, merge_lists
  import jax.numpy
  import numpy
  from jax._src.pallas import core
label_desc: Use image node_id 5 for calling a global function with example usage: _pallas_call_abstract_eval() and returns: map