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

docarray

Represent, send, store and search multimodal data

voyager

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

torchpq

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

mlpack

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

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 🔍

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.

nilvec

A lightweight library for nearest neighbor search in Rust.

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

pylmnn

Large Margin Nearest Neighbors implementation in python

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