to save a model and its configuration file to. Will be created if it doesn’t exist.   params (Union[Dict, FrozenDict]) —
A PyTree of model parameters.   is_main_process (bool, optional, defaults to True) —
Whether the process calling this is the main process or not. Useful during distributed training and you
need to call this function on all processes. In this case, set is_main_process=True only on the main
process to avoid race conditions.   p