Representation Learning Python Packages

s3prl

Self-Supervised Speech Pre-training and Representation Learning Toolkit

espnet-s3prl

Self-Supervised Speech Pre-training and Representation Learning Toolkit

augmented-pca

Repository for the AugmentedPCA Python package.

visbench

Probe any vision backbone across high-, mid- and low-level computer vision tasks

disent

🧶 Modular VAE disentanglement framework for python built with PyTorch Lightning ▸ Including metrics and datasets ▸ With strongly supervised, weakly supervised and unsupervised methods ▸ Easily configured and run with Hydra config ▸ Inspired by disentanglement_lib

hyper-models

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

glucotrace

Turn a day of continuous glucose monitor (CGM) data into a 128-number fingerprint. Small, pre-registered, research-only PyTorch encoder.

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

neembed-geoopt

Fine-tune Sentence Transformers with Geoopt for hyperbolic and mixed-curvature embeddings, semantic retrieval, and hierarchy-aware learning.

open-metric-learning

Metric learning and retrieval pipelines, models and zoo.

cellmapper

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

foundation-cancer-image-biomarker

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

som-lilypond

Intuitive visualization tool for high-dimensional data using Self-Organizing Maps

grailts

"GRAIL: Efficient Time-Series Representation Learning"

pragmatiq

Independent banking event-sequence modeling toolkit inspired by arXiv:2604.08649; not affiliated with or endorsed by Revolut.

hyperview

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

autobotlib

An autoML for explainable text classification.

simclr

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

mklpy

A package for Multiple Kernel Learning in Python

avex

Animal Vocalization Encoder Library

rs-embed

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

medtokenizers

Tokenizers for 2D and 3D medical images: VQ, LFQ, FSQ, Residual FSQ, VAE, and AE quantizers behind one backbone.

ampligraph

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

compresso-pytorch

A PyTorch Framework for Sparse Representation Learning