code: def simple_net_block(nin, nout):
      return objax.nn.Sequential(
          [
              objax.nn.Conv2D(nin, nout, k=3),
              objax.functional.leaky_relu,
              objax.functional.max_pool_2d,
              objax.nn.Conv2D(nout, nout, k=3),
              objax.functional.leaky_relu,
          ]
      )
  
file_path: objax/examples/image_classification/mnist_cnn.py
imports: import os
  import numpy
  import tensorflow_datasets
  from tqdm import trange
  import objax
  from objax.util import EasyDict
label_desc: Use image node_id 1 for calling a global function with example usage: simple_net_block(nin, nout) and returns: objax