code: def max_pool_2d(x):
      return functools.partial(
          objax.functional.max_pool_2d,
          size=2,
          strides=2,
          padding=objax.constants.ConvPadding.VALID,
      )(x)
  
file_path: objax/objax/zoo/vgg.py
imports: import functools
  import os
  from urllib import request
  import jax.numpy
  import numpy
  import objax
label_desc: Use image node_id 2 for calling a global function with example usage: max_pool_2d(x) and returns: functools