Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations" - View it on GitHub