ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Represents the state of the server carried between rounds.

model a ModelWeights structure, containing Tensors or Variables.
optimizer_state a list of Tensors or Variables, in the order returned by optimizer.variables()
delta_aggregate_state state (possibly empty) of the delta_aggregate_fn.
model_broadcast_state state (possibly empty) of the model_broadcast_fn.