n is not provided, kwargs are first passed to the configuration class
initialization function from_config(). Each key of the kwargs that corresponds
to a configuration attribute is used to override said attribute with the supplied kwargs value.
Remaining keys that do not correspond to any configuration attribute are passed to the underlying
model’s __init__ function.
    Instantiate a pretrained Flax model from a pretrained model configuration.