src/transformers/models/megatron_bert/modeling_megatron_bert.py/MegatronBertModel/__init__
class MegatronBertModel:    def __init__(self, config, add_pooling_layer=True):
        super().__init__(config)
        self.config = config

        self.embeddings = MegatronBertEmbeddings(config)
        self.encoder = MegatronBertEncoder(config)

        self.pooler = MegatronBertPooler(config) if add_pooling_layer else None

        self.init_weights()