Rectified Adam (RAdam) optimizer with Keras