code: def _initial_style_open_jaxpr(
      fun: Callable,
      in_tree,
      in_avals,
      primitive_name: str | None = None,
  ):
      wrapped_fun, out_tree = flatten_fun_nokwargs(
          lu.wrap_init(fun), in_tree
      )
      debug = pe.debug_info(
          fun, in_tree, out_tree, False, primitive_name or "<unknown>"
      )
      jaxpr, _, consts = pe.trace_to_jaxpr_dynamic(
          wrapped_fun, in_avals, debug
      )
      return jaxpr, consts, out_tree()
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  import os
  from functools import partial
  from typing import Any, Callable
  from jax._src import core
  from jax._src import linear_util
  from jax._src.lax import lax
  from jax._src import effects
  from jax._src import ad_util
  from jax._src import state
  from jax._src import util
  from jax._src.util import cache, weakref_lru_cache, safe_map, unzip3, partition_list
  from jax.api_util import flatten_fun_nokwargs
  from jax._src.interpreters import partial_eval
  from jax.tree_util import tree_map, tree_unflatten
label_desc: Use image node_id 3 for calling a global function with example usage: _initial_style_open_jaxpr(fun, in_tree, in_avals, primitive_name) and returns: jaxpr,  consts,  out_tree