def METHOD_NAME(self, param_group: Any):
    r"""Add a new param group
    """
    # TODO: implement add_param_group
    raise NotImplementedError("ShardedOptimizer add_param_group not implemented yet!")