code: def make_mlp_model():
      """Instantiates a new MLP, followed by LayerNorm.
  
      The parameters of each new MLP are not shared with others generated by
      this function.
  
      Returns:
        A Sonnet module which contains the MLP and LayerNorm.
      """
      return snt.Sequential(
          [
              snt.nets.MLP(
                  [LATENT_SIZE] * NUM_LAYERS, activate_final=True
              ),
              snt.LayerNorm(
                  axis=-1, create_offset=True, create_scale=True
              ),
          ]
      )
  
file_path: graph_nets/graph_nets/demos_tf2/models.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from graph_nets import modules
  from graph_nets import utils_tf
  from six.moves import range
  import sonnet
label_desc: Use image node_id 1 for calling a global function with example usage: make_mlp_model() and returns: snt