Optimizer

Hi, using tf.math.sqrt is destabilizing the training procedure. I have used adam optimizer. just wondering if it has to do with the optimizer or I have to use another libraray function