class_docstr: A stack of encoder layers.
code: def setup(self):
      self.t5_config = t5_network.T5Config(
          vocab_size=self.vocab_size,
          emb_dim=self.emb_dim,
          num_heads=self.num_heads,
          num_encoder_layers=self.num_encoder_layers,
          num_decoder_layers=self.num_decoder_layers,
          head_dim=self.head_dim,
          mlp_dim=self.mlp_dim,
          dropout_rate=self.dropout_rate,
          dtype=self.dtype,
          mlp_activations=self.mlp_activations,
          logits_via_embedding=self.logits_via_embedding,
      )
  
imports: import functools
  from typing import Optional, Sequence
  from flax import linen
  import jax
  import jax.numpy
  from scenic.projects.knowledge_visual_language.models import constants
  from t5x.examples.t5 import layers
  from t5x.examples.t5 import network
label_desc: Use image node_id 1 for calling the TransformerHead obj's underlying member method code with example usage: obj.setup() without return types