Module: tfm.optimization.lamb

Layer-wise Adaptive Moments (LAMB) optimizer.

See paper Large Batch Optimization for Deep Learning: Training BERT in 76 minutes.

Classes

class LAMB: Optimizer that implements the Layer-wise Adaptive Moments (LAMB).

Type Aliases

FloatTensorLike