code: def get_jaxpr_type_signature(
      jaxpr: core.Jaxpr,
      *,
      consumed_inputs: list[bool | np.ndarray] | None = None,
      use_forwarding: bool = True,
  ) -> KeyReuseSignature:
      """Parse the jaxpr to determine key reuse signature"""
      if use_forwarding:
          return _forwarding.get_jaxpr_type_signature(
              jaxpr, consumed_inputs
          )
      else:
          return _simple.get_jaxpr_type_signature(
              jaxpr, consumed_inputs
          )
  
file_path: jax/jax/experimental/key_reuse/_core.py
imports: from __future__ import annotations
  from typing import Any, Callable
  from jax import core
  from jax.experimental.key_reuse import _forwarding
  from jax.experimental.key_reuse import _simple
  import numpy
label_desc: Use image node_id 3 for calling a global function with example usage: get_jaxpr_type_signature(jaxpr) and returns: _forwarding, _simple