Retrieval Python Packages

fastembed

Fast, Accurate, Lightweight Python library to make State of the Art Embedding

mteb

MTEB: Massive Text Embedding Benchmark

bm25s

Fast BM25 search in Python, powered by Numpy and Numba

semble

Fast and Accurate Code Search for Agents. Uses ~98% fewer tokens than grep+read

beir

A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.

pageindex

📑 PageIndex: Document Index for Vectorless, Reasoning-based RAG

fastembed-gpu

Fast, Accurate, Lightweight Python library to make State of the Art Embedding

gritlm

Generative Representational Instruction Tuning

libfmp

libfmp - Python package for teaching and learning Fundamentals of Music Processing (FMP)

aelfrice

Bayesian memory that learns from feedback for LLM agents

act-atmos

Atmospheric data Community Toolkit - A python based toolkit for exploring and analyzing time series atmospheric datasets

mxbai-rerank

Crispy reranking models by Mixedbread

ctx-retriever

Trigger-Driven Dynamic Context Loading for Code-Aware LLM Agents

inferedge-moss

The retrieval layer for production AI systems. Lightning-fast (<10ms) search without vector databases. Built for browser, edge, on-device, and cloud.

rtfm-ai

The open retrieval layer for AI coding agents. Indexes code, docs, legal, research, data — 22 parsers (incl. EPUB, DOCX, ODT), FTS5 + semantic search, knowledge graph. Serves surgical context via MCP. Open source, local, free.

openkb

OpenKB: Open LLM Knowledge Base

cellin

Build long-lived multimodal memory, dream over it, and retrieve context with transparent weighting.

bm25

Fast BM25 search in Python, powered by Numpy and Numba

intel-extension-for-transformers

⚡ Build your chatbot within minutes on your favorite device; offer SOTA compression techniques for LLMs; run LLMs efficiently on Intel Platforms⚡

brinicle

A resource-efficient C++ vector index engine built for low-RAM production workloads

byaldi

Use late-interaction multi-modal models such as ColPali in just a few lines of code.

vidore-benchmark

Vision Document Retrieval (ViDoRe): Benchmark. Evaluation code for the ColPali paper.

memobase

User Profile-Based Long-Term Memory for AI Chatbot Applications.

memorizing-transformers-pytorch

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