src/transformers/models/rembert/modeling_tf_rembert.py/TFRemBertForMaskedLM/get_lm_head
class TFRemBertForMaskedLM:    def get_lm_head(self) -> tf.keras.layers.Layer:
        return self.mlm.predictions