Combining Optimizers

Combining Optimizers#

chain(*args)

Applies a list of chainable update transformations.

named_chain(*args)

Applies a list of named chainable update transformations.

partition(transforms, param_labels, *[, ...])

Partitions params and applies a different transformation to each subset.

PartitionState(inner_states)