src/transformers/models/esm/modeling_esmfold.py/EsmFoldStructureModuleTransition/forward
class EsmFoldStructureModuleTransition:    def forward(self, s):
        for l in self.layers:
            s = l(s)

        s = self.dropout(s)
        s = self.layer_norm(s)

        return s