class_docstr: Module to compose parameterized functions, implemented as a scan.
class_docstr_tok: ["Module","to","compose","parameterized","functions",",","implemented","as","a","scan","."]
code: def _call_wrapped(
      self,
      x: jax.Array,
      *args,
  ) -> tuple[jax.Array, Optional[jax.Array]]:
      raise NotImplementedError()
  
imports: import collections
  import functools
  import inspect
  from typing import Any, Callable, Optional, Protocol, Union
  from haiku._src import base
  from haiku._src import lift
  from haiku._src import module
  from haiku._src import transform
  import jax
  import jax.numpy
label_desc: Use image node_id 3 for calling the _LayerStack obj's underlying member method code with example usage: obj._call_wrapped(x) without return types