PyTorch implementation of the InfoNCE loss for self-supervised learning.
Primary LanguagePythonMIT LicenseMIT