src/transformers/models/distilbert/modeling_distilbert.py/FFN/forward
class FFN:    def forward(self, input: torch.Tensor) -> torch.Tensor:
        return apply_chunking_to_forward(self.ff_chunk, self.chunk_size_feed_forward, self.seq_len_dim, input)