Approximate Nearest Neighbors Python Packages

oasysdb

In-memory vector store with efficient read and write performance for semantic caching and retrieval system. Redis for Semantic Caching.

memorizing-transformers-pytorch

Implementation of Memorizing Transformers (ICLR 2022), attention net augmented with indexing and retrieval of memories using approximate nearest neighbors, in Pytorch

entity-embed

PyTorch library for transforming entities like companies, products, etc. into vectors to support scalable Record Linkage / Entity Resolution using Approximate Nearest Neighbors.

pupyl

🧿 Pupyl is a really fast image search library which you can index your own (millions of) images and find similar images in millisecond.