Similarity Search Python Packages

lancedb

Developer-friendly OSS embedded retrieval library for multimodal AI. Search More; Manage Less.

chonkie

🦛 CHONK docs with Chonkie ✨ — The lightweight ingestion library for fast, efficient and robust RAG pipelines

usearch

Fast Open-Source Search & Clustering engine × for Vectors & Arbitrary Objects × in C++, C, Python, JavaScript, Rust, Java, Objective-C, Swift, C#, GoLang, and Wolfram 🔍

gptcache

Semantic cache for LLMs. Fully integrated with LangChain and llama_index.

libcuvs-cu12

cuVS - a library for vector search and clustering on the GPU

zvec

A lightweight, lightning-fast, in-process vector database

cuvs-cu12

cuVS - a library for vector search and clustering on the GPU

elastiknn-client

Elasticsearch plugin for nearest neighbor search. Store vectors and run similarity search using exact and approximate algorithms.

usearch-iscc

Fast Open-Source Search & Clustering engine × for Vectors & Arbitrary Objects × in C++, C, Python, JavaScript, Rust, Java, Objective-C, Swift, C#, GoLang, and Wolfram 🔍

fpsim2

Simple package for fast molecular similarity searches

setsimilaritysearch

All-pair set similarity search on millions of sets in Python and on a laptop

oasysdb

Fast & flexible embedded vector database with incremental HNSW indexing.

fastembed-vectorstore

In-memory vector store with FastEmbed integration for Python applications.

tfsim-nightly

TensorFlow Similarity is a python package focused on making similarity learning quick and easy.

torchdr

TorchDR - PyTorch Dimensionality Reduction

pyvsag

vsag is a vector indexing library used for similarity search.

liteindex

ultra fast, thread and process safe, easily queryable Indexes/ caches for Python.

grailts

"GRAIL: Efficient Time-Series Representation Learning"

tensorflow-similarity

TensorFlow Similarity is a python package focused on making similarity learning quick and easy.

midisim

Calculate, search, and analyze MIDI-to-MIDI similarity at scale

cppbktree

Python BK-Tree module based on a C++ implementation

bertsimilar

Get Similar Words and Embeddings using BERT Models

iscc-usearch

Scalable approximate nearest neighbor search for variable-length binary bit-vectors using NPHD metric.

quaterion

Blazing fast framework for fine-tuning similarity learning models