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.

honcho-ai

Memory library for building stateful agents

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

flagembedding

Retrieval and Retrieval-augmented LLMs

model2vec

Fast State-of-the-Art Static Embeddings

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. (提供大模型工具链最佳实践，以及优雅且便捷地访问千帆大模型平台）

lightly

A python library for self-supervised learning on images.

lightly-utils

A python library for self-supervised learning on images.

vecs

Postgres/pgvector Python Client

vicinity

Lightweight Nearest Neighbors with Flexible Backends

mlx-embeddings

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

node2vec

Implementation of the node2vec algorithm.

semble

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

tf-hub-nightly

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

navec

Compact high quality word embeddings for Russian language

polyfuzz

Fuzzy string matching, grouping, and evaluation.

natasha

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

adaptive-classifier

A flexible, adaptive classification system for dynamic text classification

yantrikdb

Cognitive memory engine for AI agents — temporal decay, contradiction detection, autonomous consolidation, knowledge graph, ANN recall via HNSW. Embeddable Rust library with Python bindings; powers yantrikdb-server (HTTP gateway, MCP server, openraft cluster). AGPL.

txtai

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

lilbee

Terminal-first local search and AI chat over your documents, code, and crawled websites. Semantic + hybrid search, vision OCR, auto-built wiki, browsable GGUF model catalog. Works as CLI, TUI, MCP server, REST API, or Python library. Offline by default, no sidecar services.