code: def add_impl(x, y):
      return raw_jaxval_adders[type(x)](x, y)
  
imports: from __future__ import annotations
  import types
  from typing import Any, Callable, TypeVar
  from jax._src import core
  from jax._src import traceback_util
  from jax._src.core import Primitive, valid_jaxtype, raise_to_shaped, get_aval
  from jax._src.tree_util import register_pytree_node
  from jax._src.typing import Array, ArrayLike
  from jax._src.util import safe_map
label_desc: Use image node_id 2 for calling a global function with example usage: add_impl(x, y) and returns: raw_jaxval_adders