code: def __call__(self, /, *args, **kwargs):
      __tracebackhide__ = True
      try:
          return self._call(False, args, kwargs)
      except XlaRuntimeError as e:
          # Catch Equinox's runtime errors, and strip the more intimidating parts of
          # the error message.
          if len(e.args) != 1 or not isinstance(e.args[0], str):
              raise  # No idea if this ever happens. But if it does, just bail.
          (msg,) = e.args
          if "EqxRuntimeError: " in msg:
              _, msg = msg.split("EqxRuntimeError: ", 1)
              msg, _ = msg.rsplit("\n\nAt:\n", 1)
              msg = msg + _eqx_on_error_msg
              e.args = (msg,)
              if (
                  jax.config.jax_traceback_filtering
                  in (  # pyright: ignore
                      None,
                      "auto",
                  )
              ):
                  _modify_traceback(e)
          raise
  
imports: import functools
  import inspect
  import warnings
  from collections.abc import Callable
  from typing import Any, Literal, overload, TypeVar
  from typing_extensions import ParamSpec
  import jax
  import jax._src.traceback_util
  import jax.core
  from jaxtyping import PyTree
  from ._compile_utils import compile_cache, get_fn_names, hashable_combine, hashable_filter, hashable_partition, Lowered
  from ._custom_types import sentinel
  from ._deprecate import deprecated_0_10
  from ._doc_utils import doc_remove_args
  from ._filters import combine, is_array, partition
  from ._module import field, Module, module_update_wrapper, Partial, Static
label_desc: Use image node_id 3 for calling the _JitWrapper obj's underlying member method code with example usage: obj.__call__() and returns: self