def METHOD_NAME(self, state_dict: Mapping[str, Any]):
    r"""Loads the ShardedOptimizer state.
    Args:
        state_dict (dict): ShardedOptimizer state. Should be an object returned
            from a call to :meth:`state_dict`.
    """
    # TODO: implement load_state_dict
    raise NotImplementedError("ShardedOptimizer load_state_dict not implemented yet!")