Top 7 Python nearest-neighbor-search Projects

docarray

32 2,739 9.2 Python

Represent, send, store and search multimodal data

Project mention: DocArray – Represent, send, and store multimodal data for ML | news.ycombinator.com | 2023-04-27

similarity

7 994 6.5 Python

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

workos.com sponsored

The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
pynndescent

4 838 6.5 Python

A Python nearest neighbor descent for approximate nearest neighbors

Project mention: [D]: Best nearest neighbour search for high dimensions | /r/MachineLearning | 2023-05-17

I'll assume this is the link to pynndescent, looks cool! Thanks for sharing. I haven't used it before. Also seems like it's an approximate nearest neighbor algorithm, just FYI for others seeing this.

quaterion

4 622 2.3 Python

Blazing fast framework for fine-tuning similarity learning models
pgANN

2 289 0.0 Python

Fast Approximate Nearest Neighbor (ANN) searches with a PostgreSQL database.

Project mention: Pinecone raises $100M Series B | news.ycombinator.com | 2023-04-27

Why do you use pgvector instead of pgANN? My understanding is pgANN is built with FAISS. When I compared pgvector with FAISS, pgvector was 3-5x slower.
https://github.com/netrasys/pgANN

citrus

1 92 7.6 Python

(distributed) vector database (by 0xDebabrata)

Project mention: Created a smol vector database in my free time. Looking to provide a LangChain integration soon! | /r/LangChain | 2023-05-06

It supports all the basic features like creating an index, inserting vectors and searching through them. Here's the GitHub link if anyone's interested in going over it: https://github.com/0xDebabrata/citrus

tinyknn

1 12 6.6 Python

A tiny approximate K-Nearest Neighbour library in Python based on Fast Product Quantization and IVF
InfluxDB

www.influxdata.com sponsored

Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).