src/transformers/models/esm/modeling_esmfold.py/EsmFoldRelativePosition/__init__
class EsmFoldRelativePosition:    def __init__(self, config):
        super().__init__()
        self.bins = config.position_bins

        # Note an additional offset is used so that the 0th position
        # is reserved for masked pairs.
        self.embedding = torch.nn.Embedding(2 * self.bins + 2, config.pairwise_state_dim)