src/transformers/models/xlm_roberta/modeling_flax_xlm_roberta.py/FlaxXLMRobertaPooler/setup
class FlaxXLMRobertaPooler:    def setup(self):
        self.dense = nn.Dense(
            self.config.hidden_size,
            kernel_init=jax.nn.initializers.normal(self.config.initializer_range),
            dtype=self.dtype,
        )