An Implementation for the optimizer LAMB with Tensorflow 2.0 - View it on GitHub