Representation Learning Python Packages

s3prl

Self-Supervised Speech Pre-training and Representation Learning Toolkit

relic-pytorch

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

autobotlib

AutoBOT: Explainable AutoML for texts

grailts

"GRAIL: Efficient Time-Series Representation Learning"

crosslearn

Reusable representation extractors for reinforcement learning across vectors, images, and Chronos-backed time-series windows.

open-metric-learning

Metric learning and retrieval pipelines, models and zoo.

foundation-cancer-image-biomarker

[Nature Machine Intelligence 2024] Code and evaluation repository for the paper

hyperview

HyperView curates datasets and provides model introspection in hyperbolic and Euclidean geometries.

deberta

The implementation of DeBERTa

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 🖼️ & 🖋️

cellmapper

k-NN-based mapping of cells across representations to transfer labels, embeddings, and expression values.

topoembedx

Representation Learning on Topological Domains

ampligraph

Python library for Representation Learning on Knowledge Graphs https://docs.ampligraph.org

embedders

With embedders, you can easily convert your texts into sentence- or token-level embeddings within a few lines of code. Use cases for this include similarity search between texts, information extraction such as named entity recognition, or basic text classification.

hyper-models

Model zoo for non-Euclidean embedding models (hyperbolic, hyperspherical) - ONNX exports for HyperView

torch-molecule

Deep learning packages for molecular discovery with a simple sklearn-style interface

avex

Animal Vocalization Encoder Library

reptimeline

Track how discrete representations evolve during neural network training — lifecycle events, phase transitions, ontology discovery, and causal verification

entity-embed

PyTorch library for transforming entities like companies, products, etc. into vectors to support scalable Record Linkage / Entity Resolution using Approximate Nearest Neighbors.

disent

Vae disentanglement framework built with pytorch lightning.

simple-clip

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

rs-embed

One line code to get any remote sensing foundation model embeddings for any place and any time

pykg2vec

A Python library for Knowledge Graph Embedding

torchok

Production-oriented Computer Vision models training pipeline for common tasks: classification, segmentation, detection and representation🥤