examples/research_projects/deebert/src/modeling_highway_bert.py/DeeBertModel/__init__
class DeeBertModel:    def __init__(self, config):
        super().__init__(config)
        self.config = config

        self.embeddings = BertEmbeddings(config)
        self.encoder = DeeBertEncoder(config)
        self.pooler = BertPooler(config)

        self.init_weights()