Code release for the paper Perfectly Balanced: Improving Transfer and Robustness of Supervised Contrastive Learning - View it on GitHub