Douglas NybergandGitHub f5abd38132 remove tfa dependency: use keras.optimizers.Lamb and tf.raw_ops for LARS (#13555)

2025-12-03 17:48:27 -05:00

..

lars_optimizer.py

remove tfa dependency: use keras.optimizers.Lamb and tf.raw_ops for LARS (#13555)

2025-12-03 17:48:27 -05:00

lars_util.py

PolynomialDecayWithWarmup + tests (#3649)

2024-03-07 18:53:36 -05:00