As per [this warning](https://github.com/tensorflow/addons/blob/7aac34c52fc8fa90a2581d3bfb086ee73ef5f61e/tensorflow_addons/optimizers/rectified_adam.py#L132).
