code: def _bench_trivial_dispatch(state, key):
      f = jax.jit(lambda key: key)
      _ = f(key)
      while state:
          f(key)
      f(key).block_until_ready()
  
file_path: jax/benchmarks/random_benchmark.py
imports: import google_benchmark
  import jax
  from jax import dtypes
label_desc: Use image node_id 3 for calling a global function with example usage: _bench_trivial_dispatch(state, key) without return types