src/transformers/models/longt5/modeling_flax_longt5.py/FlaxLongT5LayerNorm/setup
class FlaxLongT5LayerNorm:    def setup(self):
        self.weight = self.param("weight", self.weight_init, (self.hidden_size,))