Contrastive Learning Python Packages

pytorch-metric-learning

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

lightly

A python library for self-supervised learning on images.

lightly-utils

A python library for self-supervised learning on images.

x-clip

A concise but complete implementation of CLIP with various experimental improvements from recent papers

cebra

Learnable latent embeddings for joint behavioral and neural analysis - Official implementation of CEBRA

tfsim-nightly

TensorFlow Similarity is a python package focused on making similarity learning quick and easy.

info-nce-pytorch

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

coca-pytorch

Implementation of CoCa, Contrastive Captioners are Image-Text Foundation Models, in Pytorch

relic-pytorch

A simple PyTorch implementation of the Representation Learning via Invariant Causal Mechanisms self-supervised contrastive learning paper

libauc

LibAUC: A Deep Learning Library for X-Risk Optimization

contrastive-rl-pytorch

Contrastive Reinforcement Learning

tensorflow-similarity

TensorFlow Similarity is a python package focused on making similarity learning quick and easy.

sc-concept

code for scConcept

uform

Pocket-Sized Multimodal AI for content understanding and generation across multilingual texts, images, and 🔜 video, up to 5x faster than OpenAI CLIP and LLaVA 🖼️ & 🖋️

group-appeal-detector

Python package to automatically detect presence and location of a social group mention in political text as well as the author's stance toward the referenced group and the qualitative group category.

ak-ssl

A Self-Supervised Learning Library

mammut-pytorch

Implementation of MaMMUT, a simple vision-encoder text-decoder architecture for multimodal tasks from Google, in Pytorch

simple-clip

A minimal, but effective implementation of CLIP (Contrastive Language-Image Pretraining) in PyTorch

quaterion

Similarity Learning fine-tuning framework

pygcl

PyGCL: A PyTorch Library for Graph Contrastive Learning

contrastive-ft

CRAFT: Contrastive Representation Aware Fine-Tuning toolkit

simclr

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.

cell-decipher

DECIPHER for learning disentangled cellular embeddings in large-scale heterogeneous spatial omics data

open-retrievals

All-in-One: Text Embedding, Retrieval, Reranking and RAG in Transformers