Pytorch code release of CVPR 21 Paper: Learning Invariant Representations and Risks - View it on GitHub