code: def consume(key):
      """Consume the key and return a consumed copy."""
      return consume_p.bind(key)
  
file_path: jax/jax/experimental/key_reuse/_common.py
imports: from __future__ import annotations
  from typing import NamedTuple
  from jax import core
  from jax.interpreters import batching, mlir
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: consume(key) and returns: consume_p