code: def instantiate(z: Zero | Array) -> Array:
      if isinstance(z, Zero):
          return zeros_like_aval(z.aval)
      return z
  
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 6 for calling a global function with example usage: instantiate(z) and returns: z, zeros_like_aval