src/transformers/modeling_flax_utils.py/FlaxPreTrainedModel/init_weights
class FlaxPreTrainedModel:    def init_weights(self, rng: jax.random.PRNGKey, input_shape: Tuple, params: FrozenDict = None) -> Dict:
        raise NotImplementedError(f"init method has to be implemented for {self}")