code: def check_key_reuse(
      fun: Callable[..., Any],
      /,
      *args: Any,
      use_forwarding: bool = True,
  ) -> KeyReuseSignature:
      """Function to statically check key reuse."""
      if use_forwarding:
          return _forwarding.check_key_reuse(fun, *args)
      else:
          return _simple.check_key_reuse(fun, *args)
  
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 1 for calling a global function with example usage: check_key_reuse() and returns: _forwarding, _simple