src/transformers/models/convnext/modeling_convnext.py/ConvNextStage/forward
class ConvNextStage:    def forward(self, hidden_states: torch.FloatTensor) -> torch.Tensor:
        hidden_states = self.downsampling_layer(hidden_states)
        hidden_states = self.layers(hidden_states)
        return hidden_states