# Implementation of LAMB in Tensorflow: https://arxiv.org/pdf/1904.00962.pdf - View it on GitHub