src/transformers/models/big_bird/modeling_flax_big_bird.py/FlaxBigBirdEncoder/setup
class FlaxBigBirdEncoder:    def setup(self):
        self.layer = FlaxBigBirdLayerCollection(
            self.config,
            dtype=self.dtype,
            gradient_checkpointing=self.gradient_checkpointing,
        )