code: def _assert_typed_key(key):
      assert dtypes.issubdtype(key.dtype, dtypes.prng_key)
  
file_path: jax/benchmarks/random_benchmark.py
imports: import google_benchmark
  import jax
  from jax import dtypes
label_desc: Use image node_id 2 for calling a global function with example usage: _assert_typed_key(key) without return types