PyTorch implementation of the InfoNCE loss for self-supervised learning. - info-nce-pytorch/info_nce/__init__.py at main · RElbers/info-nce-pytorch