The public API for contributors who develop federated learning algorithms.
Represents a client computation that produces an update to a model.
class ClientOutput: Structure
for outputs returned from clients during federated optimization.
class EnhancedModel: A
wrapper around a Model that adds sanity checking and metadata helpers.
A wrapper around a Model that adds sanity checking and metadata helpers.
class ModelWeights: A
container for the trainable and non-trainable variables of a
StatefulBroadcastFn for weights of model returned by