class_docstr: A multi-layer perceptron.
class_docstr_tok: ["A","multi-layer","perceptron","."]
code: def __call__(self, inputs):
      x = inputs
      for size in self.feature_sizes:
          x = nn.Dense(features=size)(x)
          x = self.activation(x)
          x = nn.Dropout(
              rate=self.dropout_rate, deterministic=self.deterministic
          )(x)
      return x
  
file_path: jax/jax/experimental/jax2tf/tests/flax_models/gnn.py
imports: from collections.abc import Sequence
  from typing import Callable
  from flax import linen
  import jax
  import jax.numpy
  import jraph
label_desc: Use image node_id 1 for calling the MLP obj's underlying member method code with example usage: obj.__call__(inputs) and returns: x