class_docstr: Simple MLP for Transformer.
class_docstr_tok: ["Simple","MLP","for","Transformer","."]
code: def __call__(self, x: jnp.ndarray) -> jnp.ndarray:
      ch = x.shape[-1]
      x = nn.Dense(4 * ch, name="c_fc")(x)
      x = quick_gelu(x)
      x = nn.Dense(ch, name="c_proj")(x)
      return x
  
imports: import functools
  from typing import Sequence, Optional, Union, Tuple
  import flax.linen
  import jax
  import jax.numpy
label_desc: Use image node_id 1 for calling the MLP obj's underlying member method code with example usage: obj.__call__(x) and returns: x