src/transformers/models/big_bird/modeling_flax_big_bird.py/FlaxBigBirdLayerCollection/setup
class FlaxBigBirdLayerCollection:    def setup(self):
        self.layers = [
            FlaxBigBirdLayer(self.config, layer_id=i, name=str(i), dtype=self.dtype)
            for i in range(self.config.num_hidden_layers)
        ]