code: def custom_container_raw_key(state):
      key = jax.random.PRNGKey(0)
      _assert_raw_key(key)
      _bench_custom_container(state, key)
  
file_path: jax/benchmarks/random_benchmark.py
imports: import google_benchmark
  import jax
  from jax import dtypes
label_desc: Use image node_id 12 for calling a global function with example usage: custom_container_raw_key(state) without return types