examples/research_projects/performer/modeling_flax_performer.py/FlaxPerformerEncoder/__call__
class FlaxPerformerEncoder:    def __call__(self, hidden_state, attention_mask):
        layer = FlaxPerformerLayerCollection(
            self.num_layers,
            self.num_heads,
            self.head_size,
            self.intermediate_size,
            name="layer",
            hidden_act=self.hidden_act,
        )(hidden_state, attention_mask)
        return layer