src/transformers/models/levit/modeling_levit.py/LevitMLPLayer/forward
class LevitMLPLayer:    def forward(self, hidden_state):
        hidden_state = self.linear_up(hidden_state)
        hidden_state = self.activation(hidden_state)
        hidden_state = self.linear_down(hidden_state)
        return hidden_state