src/transformers/models/bit/modeling_bit.py/BitStage/forward
class BitStage:    def forward(self, input: Tensor) -> Tensor:
        hidden_state = input
        for _, layer in enumerate(self.layers):
            hidden_state = layer(hidden_state)
        return hidden_state