info-nce-pytorch

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

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install info-nce-pytorch

Dependencies

Releases

Version Released Bookworm
Python 3.11
Trixie
Python 3.13
Files
0.1.4 2021-10-23
0.1.3 2021-05-25
0.1.1 2021-05-21
0.1.0 2021-05-13

Issues with this package?

Page last updated 2026-09-03 00:46:00 UTC