src/transformers/models/esm/modeling_esmfold.py/EsmFoldBackboneUpdate/__init__
class EsmFoldBackboneUpdate:    def __init__(self, config):
        super().__init__()

        self.linear = EsmFoldLinear(config.sequence_dim, 6, init="final")