src/transformers/models/gpt_neo/modeling_flax_gpt_neo.py/FlaxGPTNeoBlockCollection/setup
class FlaxGPTNeoBlockCollection:    def setup(self):
        self.blocks = [
            FlaxGPTNeoBlock(self.config, layer_id=i, name=str(i), dtype=self.dtype)
            for i in range(self.config.num_hidden_layers)
        ]