Official repository for "Scaling Retrieval-Based Langauge Models with a Trillion-Token Datastore".
β226Dec 16, 2025Updated 7 months ago
Alternatives and similar repositories for retrieval-scaling
Users that are interested in retrieval-scaling are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Python package for serving a local search engine. One command to download and serve a datastore---that's it π.β26Jun 6, 2025Updated last year
- [ICLR 2025] BRIGHT: A Realistic and Challenging Benchmark for Reasoning-Intensive Retrievalβ210Sep 13, 2025Updated 10 months ago
- [SIGIR 2025] The official repo for "Scaling Sparse and Dense Retrieval in Decoder-Only LLMs"β22Mar 31, 2025Updated last year
- Official repository for paper "ReasonIR Training Retrievers for Reasoning Tasks".β230Jul 2, 2026Updated 3 weeks ago
- FlexAttention w/ FlashAttention3 Supportβ27Oct 5, 2024Updated last year
- Managed Kubernetes at scale on DigitalOcean β’ AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Resolving Knowledge Conflicts in Large Language Models, COLM 2024β18Oct 7, 2025Updated 9 months ago
- Expand -> Retrieve -> Rerank - simple method with strong results on BRIGHT benchmarkβ22Aug 22, 2025Updated 11 months ago
- RankLLM is a Python toolkit for reproducible information retrieval research using rerankers, with a focus on listwise reranking.β611Jul 19, 2026Updated last week
- β19Nov 4, 2025Updated 8 months ago
- Code and model release for the paper "Task-aware Retrieval with Instructions" by Asai et al.β168Oct 4, 2023Updated 2 years ago
- Scalable training for dense retrieval models.β298Jul 2, 2026Updated 3 weeks ago
- [EMNLP 2022] This is the code repo for our EMNLPβ22 paper "COCO-DR: Combating Distribution Shifts in Zero-Shot Dense Retrieval with Contrβ¦β51Oct 12, 2023Updated 2 years ago
- [ACL'24 Oral] Analysing The Impact of Sequence Composition on Language Model Pre-Trainingβ24Aug 18, 2024Updated last year
- QLoRA for Masked Language Modelingβ23Sep 11, 2023Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer β’ AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- train with kittens!β67Oct 25, 2024Updated last year
- SILO Language Models code repositoryβ83Feb 23, 2024Updated 2 years ago
- Retrieval-Augmented Generation battle!β66Apr 18, 2026Updated 3 months ago
- β11Feb 9, 2024Updated 2 years ago
- Dense X Retrieval: What Retrieval Granularity Should We Use?β171Jan 8, 2024Updated 2 years ago
- This repository provides an original implementation of Detecting Pretraining Data from Large Language Models by *Weijia Shi, *Anirudh Ajiβ¦β243Nov 3, 2023Updated 2 years ago
- [ICLR'25] Data and code for our paper "Why Does the Effective Context Length of LLMs Fall Short?"β82Nov 25, 2024Updated last year
- The original implementation of Min et al. "Nonparametric Masked Language Modeling" (paper https//arxiv.org/abs/2212.01349)β159Jan 6, 2023Updated 3 years ago
- XTR/WARP (SIGIR'25) is an extremely fast and accurate retrieval engine based on Stanford's ColBERTv2/PLAID and Google DeepMind's XTR.β211May 3, 2025Updated last year
- Managed hosting for WordPress and PHP on Cloudways β’ AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Repo for training MLMs, CLMs, or T5-type models on the OLM pretraining data, but it should work with any hugging face text dataset.β98Feb 9, 2023Updated 3 years ago
- code for training & evaluating Contextual Document Embedding modelsβ207May 14, 2025Updated last year
- High-Performance Engine for Multi-Vector Searchβ271May 28, 2026Updated 2 months ago
- [AAAI26]: DS SERVE: The Largest Open Vector Store over Pretain Data; A Framework for Efficient and Scalable Neural Retrievalβ53Jan 28, 2026Updated 6 months ago
- [COLM'25] A Controlled Study on Long Context Extension and Generalization in LLMsβ65Mar 9, 2026Updated 4 months ago
- Tevatron - Unified Document Retrieval Toolkit across Scale, Language, and Modality. Demo in SIGIR 2023, SIGIR 2025.β743Jul 18, 2026Updated last week
- Generative Representational Instruction Tuningβ697Jun 25, 2025Updated last year
- FollowIR: Evaluating and Teaching Information Retrieval Models to Follow Instructionsβ56Jul 3, 2024Updated 2 years ago
- [ACL 2024 Oral] This is the code repo for our ACLβ24 paper "MARVEL: Unlocking the Multi-Modal Capability of Dense Retrieval via Visual Moβ¦β39Jun 30, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer β’ AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Implementation of NAACL'25 "Empowering Retrieval-based Conversational Recommendation with Contrasting User Preferences"β14Sep 9, 2025Updated 10 months ago
- Source-to-Source Debuggable Derivatives in Pure Pythonβ15Jan 23, 2024Updated 2 years ago
- Learning to Retrieve by Trying - Source code for Grounding by Trying: LLMs with Reinforcement Learning-Enhanced Retrievalβ52Oct 31, 2024Updated last year
- Code and data of the EMNLP 2022 Main Conference paper "Reduce Catastrophic Forgetting of Dense Retrieval Training with Teleportation Negaβ¦β18Mar 25, 2024Updated 2 years ago
- LOFT: A 1 Million+ Token Long-Context Benchmarkβ237Apr 13, 2026Updated 3 months ago
- Document Ranking with Large Language Models.β210Feb 14, 2026Updated 5 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Trainingβ223Aug 19, 2024Updated last year