PyTorch implementation of IRMAE https//arxiv.org/abs/2010.00679 - View it on GitHub