Nearest Neighbor Search Python Packages

faiss-cpu

A library for efficient similarity search and clustering of dense vectors.

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

A library for efficient similarity search and clustering of dense vectors.

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

lynsedb

Embedded-first vector database that grows into a self-hosted service and lightweight cluster.

tiledb-vector-search

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

kannolo

Official repository of the kANNolo library.

elastiknn-client

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

mlpack

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

torchpq

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

annoy-mm

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

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 🔍

tensorflow-similarity

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

tfsim-nightly

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

liteindex

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

nilvec

A lightweight library for nearest neighbor search in Rust.

nanopq

Pure python implementation of product quantization for nearest neighbor search

squarenet

Gridification for point clouds in any dimension (fast greedy optimal transport through Cartesian Sort).

pylmnn

Large Margin Nearest Neighbor implementation in python