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

Module: tff.learning.optimizers

Libraries for optimization algorithms.

Classes

class Optimizer: Represents an optimizer for use in TensorFlow Federated.

Functions

build_sgdm(...): Returns a tff.learning.optimizers.Optimizer for momentum SGD.