Imported from https://github.com/yashkant/Padam-Tensorflow