src/transformers/models/megatron_bert/modeling_megatron_bert.py/MegatronBertOnlyMLMHead/forward
class MegatronBertOnlyMLMHead:    def forward(self, sequence_output: torch.Tensor) -> torch.Tensor:
        prediction_scores = self.predictions(sequence_output)
        return prediction_scores