code: def make_net():
      return objax.nn.Sequential(
          [
              objax.nn.Linear(1, 40),
              objax.functional.relu,
              objax.nn.Linear(40, 40),
              objax.functional.relu,
              objax.nn.Linear(40, 1),
          ]
      )
  
file_path: objax/examples/maml/maml.py
imports: import jax.numpy
  import matplotlib.pyplot
  import numpy
  from tqdm import trange
  import objax
label_desc: Use image node_id 2 for calling a global function with example usage: make_net() and returns: objax