class_docstr: Mixer model for multi-label classification task.
class_docstr_tok: ["Mixer","model","for","multi-label","classification","task","."]
code: def build_flax_model(self) -> nn.Module:
      return Mixer(
          num_classes=self.dataset_meta_data["num_classes"],
          patch_size=self.config.model.patch_size,
          hidden_size=self.config.model.hidden_size,
          num_layers=self.config.model.num_layers,
          channels_mlp_dim=self.config.model.channels_mlp_dim,
          sequence_mlp_dim=self.config.model.sequence_mlp_dim,
          dropout_rate=self.config.model.get("dropout_rate", 0.1),
          stochastic_depth=self.config.model.get(
              "stochastic_depth", 0.0
          ),
          layer_scale=self.config.model.get("layer_scale", None),
      )
  
file_path: scenic/scenic/projects/baselines/mixer.py
imports: from typing import Optional, Sequence
  import flax.linen
  import jax.numpy
  import ml_collections
  from scenic.model_lib.base_models.multilabel_classification_model import MultiLabelClassificationModel
  from scenic.model_lib.layers import attention_layers
  from scenic.model_lib.layers import nn_layers
label_desc: Use image node_id 1 for calling the MixerMultiLabelClassificationModel obj's underlying member method code with example usage: obj.build_flax_model() and returns: Mixer