class_docstr: Transformer MLP / feed-forward block.
class_docstr_tok: ["Transformer","MLP","\/","feed-forward","block","."]
docstr: Applies Transformer MlpBlock module.
docstr_tok: ["Applies","Transformer","MlpBlock","module","."]
code: def __call__(self, x, deterministic=True):
      """Applies Transformer MlpBlock module."""
      inits = dict(
          kernel_init=nn.initializers.xavier_uniform(),
          bias_init=nn.initializers.normal(stddev=1e-6),
      )
  
      n, l, d = x.shape  # pylint: disable=unused-variable
      x = jnp.array(x, dtype=self.dtype)
      x = nn.Dense(self.mlp_dim or 4 * d, **inits, dtype=self.dtype)(x)
      x = nn.gelu(x)
      x = nn.Dropout(rate=self.dropout)(x, deterministic)
      x = nn.Dense(d, **inits, dtype=self.dtype)(x)
      return x
  
file_path: scenic/scenic/projects/knowledge_visual_language/models/vit.py
imports: from typing import Optional, Sequence, Union
  from big_vision.models import vit
  import flax.linen
  import jax
  import jax.numpy
label_desc: Use image node_id 1 for calling the MlpBlock obj's underlying member method code with example usage: obj.__call__(x, deterministic) and returns: x