src/transformers/models/esm/modeling_esmfold.py/EsmFoldStructureModuleTransitionLayer/forward
class EsmFoldStructureModuleTransitionLayer:    def forward(self, s):
        s_initial = s
        s = self.linear_1(s)
        s = self.relu(s)
        s = self.linear_2(s)
        s = self.relu(s)
        s = self.linear_3(s)

        s = s + s_initial

        return s