Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch - View it on GitHub