|View source on GitHub|
Libraries for developing Federated Learning algorithms.
class ClientDeltaFn: 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.
class ModelWeights: A container for the trainable and non-trainable variables of a
class ServerState: Represents the state of the server carried between rounds.
MeasuredProcess for weights of model returned by
build_stateless_mean(...): Builds a
MeasuredProcess that wraps
parameter_count_from_model(...): Computes count of trainable parameters for a