code: def loss(x, y):
      return ((y - net(x)) ** 2).mean()
  
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 3 for calling a global function with example usage: loss(x, y) and returns: unknown