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

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

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

node2vec

Implementation of the node2vec algorithm.

nomic

Nomic Developer API SDK

polyfuzz

Fuzzy string matching, grouping, and evaluation.

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

tf-hub-nightly

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

natasha

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

txtai

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

adaptive-classifier

A flexible, adaptive classification system for dynamic text classification

tqdb

Embedded vector database in Rust with Python bindings — TurboQuant algorithm (arXiv:2504.19874), zero training, 2–4 bit compression, HNSW ANN search, WAL persistence

esperanto

A unified interface for various AI model providers