Custom optimizer keras. A class for Tensorflow specific optimizer logic.