code: def unwrap(module):
      while isinstance(module, Wrapped):
          module = module.wrapped
      return module
  
imports: from collections.abc import Sequence
  from typing import Any, Callable, NamedTuple
  import haiku
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: unwrap(module) and returns: module