Dense Retrieval Python Packages

spladerunner

Lite weight wrapper for the independent implementation of SPLADE++ models for search & retrieval pipelines. Models and Library created by Prithivi Da, For PRs and Collaboration checkout the readme.

retriv

A Python Search Engine for Humans 🥸

m3serve

Optimised BAAI/bge-m3 serving with dense + sparse + ColBERT embeddings, async dynamic batching and pipeline GPU inference

tevatron

Tevatron - Unified Document Retrieval Toolkit across Scale, Language, and Modality. Demo in SIGIR 2023, SIGIR 2025.

imrnns

Interpretable Modular Retrieval Neural Networks (IMRNNs) is a lightweight framework that augments any dense retriever with dynamic, bidirectional modulation at inference time. This iterative modulation process enables the model to adapt representations dynamically and expose interpretable semantic dependencies between queries and documents

freshstack

This repository helps you evaluate your models on the FreshStack benchmark!

distfuse

A library to calculate similarity scores between two collections of text sequences encoded using transformer models for bitext mining, dense retrieval, retrieval-based classification, and retrieval-augmented generation (RAG).

labelbank

Retrieve + rerank over a closed label bank: LLM bi-encoders with self-mined hard negatives and a generative listwise reranker. Generalized from a silver-medal solution to Kaggle Eedi — Mining Misconceptions in Mathematics.

ir-trove

A Flexible Toolkit for Dense Retrieval

rocketqa

🚀 RocketQA, dense retrieval for information retrieval and question answering, including both Chinese and English state-of-the-art models.