code: def check_key_reuse_jaxpr(
      jaxpr: core.Jaxpr, *, use_forwarding: bool = True
  ):
      """Check the jaxpr for key reuse."""
      get_jaxpr_type_signature(jaxpr, use_forwarding=use_forwarding)
  
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 2 for calling a global function with example usage: check_key_reuse_jaxpr(jaxpr) without return types