Nearest Neighbor Search Python Packages

faiss-cpu

Community-maintained faiss wheel builder

lancedb

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

pynndescent

A Python nearest neighbor descent for approximate nearest neighbors

annoy

Approximate Nearest Neighbors in C++/Python optimized for memory usage and loading/saving to disk

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 🔍

faiss-gpu

Community-maintained faiss wheel builder

voyager

🛰️ An approximate nearest-neighbor search library for Python and Java with a focus on ease of use, simplicity, and deployability.

docarray

Represent, send, store and search multimodal data

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 🔍

elastiknn-client

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

torchpq

Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda

mlpack

mlpack: a fast, header-only C++ machine learning library

tiledb-vector-search

Cloud-native vector similarity search and storage with efficient, serverless scale-out

annoy-mm

Approximate Nearest Neighbors in C++/Python optimized for memory usage and loading/saving to disk

kannolo

Official repository of the kANNolo library.

tfsim-nightly

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

ngt

Nearest Neighbor Search with Neighborhood Graph and Tree for High-dimensional Data

liteindex

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

tensorflow-similarity

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

nanopq

Pure python implementation of product quantization for nearest neighbor search

nilvec

A lightweight library for nearest neighbor search in Rust.

pylmnn

Large Margin Nearest Neighbors implementation in python

iscc-usearch

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

lynsedb

A pure Python-implemented, lightweight, server-optional, multi-end compatible, vector database deployable locally or remotely.