How to load ServerState.optimizer_state to continue training in Tensorflow Federated

Question

Does TFF have any way to save and load optimizer state similar to model weights. For model weights there are ModelWeights.assign_weights_to() and tff.lear