code: def mlp(x):
      return hk.nets.MLP([300, 100, 10])(x)
  
imports: import google_benchmark
  import haiku
  import jax
  import jax.numpy
label_desc: Use image node_id 2 for calling a global function with example usage: mlp(x) and returns: hk