code: def named_call(f=None, name=None):
      """Adds a name to a function for profiling purposes."""
      if f is None:
          return functools.partial(named_call, name=name)
  
      return jax.named_call(f, name=name)
  
imports: import functools
  import jax
  from jax import lax
  from jax import random
  from jax import tree_util
  import jax.numpy
label_desc: Use image node_id 20 for calling a global function with example usage: named_call(f, name) and returns: jax, functools