code: def _interpret_subtrace(main, *in_vals):
      trace = DotTrace(main, jax.core.cur_sublevel())
      in_tracers = [DotTracer(trace, val) for val in in_vals]
      outs = yield in_tracers, {}
      out_tracers = map(trace.full_raise, outs)
      out_vals = [t.val for t in out_tracers]
      yield out_vals
  
imports: import collections
  import contextlib
  import functools
  import html
  from typing import Any, Callable, NamedTuple, Optional
  from haiku._src import data_structures
  from haiku._src import module
  from haiku._src import utils
  import jax
  import jax.core
  from jax.experimental import pjit
  from jax.extend import linear_util
label_desc: Use image node_id 5 for calling a global function with example usage: _interpret_subtrace(main) without return types