src/transformers/models/roberta/modeling_flax_roberta.py/FlaxRobertaEncoder/setup
class FlaxRobertaEncoder:    def setup(self):
        self.layer = FlaxRobertaLayerCollection(self.config, dtype=self.dtype)