class_docstr: MTV model for multi-classification tasks.
  
  When methods are overriden by both parents, the implementation follows the
  first parent, which is ViViTMultiHeadClassificationModel in this case. For
  build_flax_model() and default_flax_model_config(), we explicitly call the
  methods from MTVClassificationModel.
code: def build_flax_model(self) -> nn.Module:
      return MTVClassificationModel.build_flax_model(self)
  
imports: import functools
  from typing import Any, List, Sequence, Tuple, Union
  import flax.linen
  import jax
  import jax.numpy
  import ml_collections
  from scenic.model_lib.layers import attention_layers
  from scenic.model_lib.layers import nn_layers
  from scenic.projects.baselines import vit
  from scenic.projects.mtv import model_utils
  from scenic.projects.vivit import model
  from scenic.train_lib import train_utils
label_desc: Use image node_id 1 for calling the MTVMultiheadClassificationModel obj's underlying member method code with example usage: obj.build_flax_model() and returns: MTVClassificationModel