src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py/FlaxWav2Vec2StableLayerNormEncoder/setup
class FlaxWav2Vec2StableLayerNormEncoder:    def setup(self):
        self.pos_conv_embed = FlaxWav2Vec2PositionalConvEmbedding(self.config, dtype=self.dtype)
        self.layer_norm = nn.LayerNorm(epsilon=self.config.layer_norm_eps, dtype=self.dtype)
        self.dropout = nn.Dropout(rate=self.config.hidden_dropout)
        self.layers = FlaxWav2Vec2EncoderLayerStableLayerNormCollection(self.config, dtype=self.dtype)