Representation Learning Python Packages

s3prl

Self-Supervised Speech Pre-training and Representation Learning Toolkit

30K 3K 535

open-metric-learning

Metric learning and retrieval pipelines, models and zoo.

6K 992 77

ampligraph

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

4K 2K 254

autobotlib

An autoML for explainable text classification.

3K 12 4

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

2K 1K 78

deberta

The implementation of DeBERTa

2K 2K 237

cellmapper

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

2K 16 4

augmented-pca

Repository for the AugmentedPCA Python package.

2K 13 0

hyperview

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

2K 20 3

hyper-models

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

2K 3 0

grailts

"GRAIL: Efficient Time-Series Representation Learning"

2K 40 40

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.

2K 21 2

pragmatiq

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

1K 32 5

avex

Animal Vocalization Encoder Library

1K 37 4

foundation-cancer-image-biomarker

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

1K 133 18

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

1K 135 18

entity-embed

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

1K 161 16

topoembedx

Representation Learning on Topological Domains

1K 97 15

torch-molecule

torch-molecule is a deep learning package for molecular discovery, designed with an sklearn-style interface for property prediction, inverse design and representation learning.

993 321 39