cerebras.modelzoo.common.optim_utils#

Helper functions related to configuration of the Optimizer.

Functions

configure_param_groups

Groups the optimizer parameters into non-overlapping groups.

flatten_optimizer_params

Config classes package optimizer related params in a sub dict.

partition_params_group_with_tags

Splits a single param group into multiple param groups based on glob patterns found in optimizer_params.