Embeddings Python Packages

fastembed

Fast, Accurate, Lightweight Python library to make State of the Art Embedding

pytorch-metric-learning

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

mteb

MTEB: State-of-the-art evaluation of embeddings across languages and modalities

tensorflow-hub

A library for transfer learning by reusing parts of TensorFlow models.

daft

High-performance data engine for AI and multimodal workloads. Process images, audio, video, and structured data at any scale

honcho-ai

Memory library for building stateful agents

model2vec

Fast State-of-the-Art Static Embeddings

flagembedding

Retrieval and Retrieval-augmented LLMs

instructorembedding

[ACL 2023] One Embedder, Any Task: Instruction-Finetuned Text Embeddings

qianfan

Provide best practices for LMOps, as well as elegant and convenient access to the features of the Qianfan MaaS Platform. (提供大模型工具链最佳实践，以及优雅且便捷地访问千帆大模型平台）

vicinity

Lightweight Nearest Neighbors with Flexible Backends

lightly

A python library for self-supervised learning on images.

lightly-utils

A python library for self-supervised learning on images.

semble

Fast and Accurate Code Search for Agents. Uses ~98% fewer tokens than grep+read

vecs

Postgres/pgvector Python Client

mlx-embeddings

MLX-Embeddings is the best package for running Vision and Language Embedding models locally on your Mac using MLX.

navec

Compact high quality word embeddings for Russian language

natasha

Solves basic Russian NLP tasks, API for lower level Natasha projects

polyfuzz

Fuzzy string matching, grouping, and evaluation.

node2vec

Implementation of the node2vec algorithm.

turbovec

A vector index built on TurboQuant, written in Rust with Python bindings

nomic

Nomic Developer API SDK

txtai

💡 All-in-one AI framework for semantic search, LLM orchestration and language model workflows

esperanto

A unified interface for various AI model providers