code: def resnet_50(x):
      return hk.nets.ResNet50(num_classes=10)(
          x, is_training=True, test_local_stats=True
      )
  
imports: import google_benchmark
  import haiku
  import jax
  import jax.numpy
label_desc: Use image node_id 3 for calling a global function with example usage: resnet_50(x) and returns: hk