src/transformers/models/albert/modeling_albert.py/AlbertLayer/ff_chunk
class AlbertLayer:    def ff_chunk(self, attention_output: torch.Tensor) -> torch.Tensor:
        ffn_output = self.ffn(attention_output)
        ffn_output = self.activation(ffn_output)
        ffn_output = self.ffn_output(ffn_output)
        return ffn_output