Dense Retrieval Python Packages

m3serve

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

retriv

A Python Search Engine for Humans 🥸

tevatron

Tevatron: A toolkit for learning and running deep dense retrieval models.

spladerunner

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

freshstack

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

distfuse

Compute DistFuse similarity scores from embedding models and APIs

rocketqa

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

ir-trove

A Flexible Toolkit for Dense Retrieval