src/transformers/models/albert/modeling_tf_albert.py/TFAlbertLayerGroup/__init__
class TFAlbertLayerGroup:    def __init__(self, config: AlbertConfig, **kwargs):
        super().__init__(**kwargs)

        self.albert_layers = [
            TFAlbertLayer(config, name=f"albert_layers_._{i}") for i in range(config.inner_group_num)
        ]