info-nce-pytorch (pypi): PyTorch implementation of the InfoNCE loss for self-supervised learning.
Keywords: contrastive-learning|contrastive-loss|self-supervised-learning