A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network - View it on GitHub