class_docstr: A simple MLP model. initialized with ones for testing weight decay.
class_docstr_tok: ["A","simple","MLP","model",".","initialized","with","ones","for","testing","weight","decay","."]
code: def __call__(self, x, train=None, debug=None):
      x = nn.Dense(
          features=3,
          bias_init=nn.initializers.ones,
          kernel_init=nn.initializers.ones,
      )(x)
      x = nn.relu(x)
      x = nn.Dense(
          features=1,
          bias_init=nn.initializers.ones,
          kernel_init=nn.initializers.ones,
      )(x)
      return x
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import flax
  import flax.linen
  import jax
  import jax.numpy
  import ml_collections
  import optax
  from scenic.train_lib import optimizers
  from scenic.train_lib import train_utils
  import tensorflow
label_desc: Use image node_id 1 for calling the MLP obj's underlying member method code with example usage: obj.__call__(x, train, debug) and returns: x