LAMB Optimizer for Large Batch Training (TensorFlow version) - View it on GitHub