GitHub repository: seanswyi/supervised-contrastive-loss
Description: PyTorch implementation of "Supervised Contrastive Learning"  (and SimCLR incidentally)