code: def __init__(self, input_size, sizes, name=None):
      super(_MLP, self).__init__(name=name)
      self.input_shape = (None, input_size)
      self.layers = []
      with self.name_scope:
          for size in sizes:
              self.layers.append(
                  _Dense(input_dim=input_size, output_size=size)
              )
              input_size = size
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax import numpy
  import neural_tangents
  from neural_tangents import experimental
  import numpy
  import tensorflow
label_desc: Use image node_id 1 to create a new _MLP object from inherited base classes: tf with example: obj = _MLP(input_size, sizes, name)