Save the date! Google I/O returns May 18-20 Register now

ParameterGroupOptimizerBuilder

public struct ParameterGroupOptimizerBuilder

Builds a ParameterGroupOptimizer. This is used at essentially the level of a single weight in the model. A mapping from parameter groups selected by ([Bool] to ParameterGroupOptimizer) defines the final optimizer.