LAMB: Layer-wise Adaptive Large-Batch Optimisation

hard
Unit 22 · Optimizers0/1 to milestone

LAMB trains BERT with batch sizes of 64k by scaling each layer's Adam update by a trust ratio.

Pro problem

Unlock this and every premium problem across all tracks — plus hidden-test grading, advanced AI engineering modules and new content every week.

Start 7-day free trial
solution.py · Python 3.13
Run your code to see results. ⌘↵ to run · ⌘⇧↵ to submit
LAMB: Layer-wise Adaptive Large-Batch Optimisation · TechGradify