Nearest Neighbor Search Python Packages

faiss-cpu

Community-maintained faiss wheel builder

16.3M 390 45

lancedb

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

7M 10K 868

pynndescent

A Python nearest neighbor descent for approximate nearest neighbors

6.9M 964 107

annoy

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

1.3M 14K 1K

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 🔍

549K 4K 311

faiss-gpu

Community-maintained faiss wheel builder

505K 390 45

docarray

Represent, send, store and search multimodal data

144K 3K 241

voyager

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

128K 2K 79

torchpq

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

22K 232 23

elastiknn-client

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

22K 393 50

mlpack

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

21K 6K 2K

tiledb-vector-search

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

12K 68 11

annoy-mm

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

8K 14K 1K

tfsim-nightly

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

6K 1K 105

kannolo

Official repository of the kANNolo library.

6K 55 8

ngt

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

5K 1K 129

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 🔍

4K 4K 311