Official repository for "Scaling Retrieval-Based Langauge Models with a Trillion-Token Datastore".
β225Dec 16, 2025Updated 10 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
- Official repository for paper "ReasonIR Training Retrievers for Reasoning Tasks".β230Jul 2, 2026Updated 4 months ago
- [ICLR 2025] BRIGHT: A Realistic and Challenging Benchmark for Reasoning-Intensive Retrievalβ219Sep 13, 2025Updated last year
- β14Jun 25, 2025Updated last year
- Scalable training for dense retrieval models.β297Jul 2, 2026Updated 4 months 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.
- Retrieval-Augmented Generation battle!β66Apr 18, 2026Updated 6 months ago
- Resolving Knowledge Conflicts in Large Language Models, COLM 2024β18Oct 7, 2025Updated last year
- [SIGIR 2025] The official repo for "Scaling Sparse and Dense Retrieval in Decoder-Only LLMs"β21Mar 31, 2025Updated last year
- [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 3 years ago
- FlexAttention w/ FlashAttention3 Supportβ27Oct 5, 2024Updated 2 years ago
- Official code and dataset for our paper: RefineBench: Evaluating Refinement Capability of Language Models via Checklistsβ16Dec 1, 2025Updated 11 months ago
- RankLLM is a Python toolkit for reproducible information retrieval research using rerankers, with a focus on listwise reranking.β682Oct 30, 2026Updated last week
- Dense X Retrieval: What Retrieval Granularity Should We Use?β172Jan 8, 2024Updated 2 years ago
- Expand -> Retrieve -> Rerank - simple method with strong results on BRIGHT benchmarkβ22Aug 22, 2025Updated last year
- Simple, predictable pricing with DigitalOcean hosting β’ AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Generative Representational Instruction Tuningβ702Jun 25, 2025Updated last year
- Tevatron - Unified Document Retrieval Toolkit across Scale, Language, and Modality. Demo in SIGIR 2023, SIGIR 2025.β751Oct 24, 2026Updated 2 weeks ago
- Code and model release for the paper "Task-aware Retrieval with Instructions" by Asai et al.β168Oct 4, 2023Updated 3 years ago
- SILO Language Models code repositoryβ83Feb 23, 2024Updated 2 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.β223May 3, 2025Updated last year
- [ICLR'25] Data and code for our paper "Why Does the Effective Context Length of LLMs Fall Short?"β83Nov 25, 2024Updated last year
- FollowIR: Evaluating and Teaching Information Retrieval Models to Follow Instructionsβ58Jul 3, 2024Updated 2 years ago
- ScienceMeter: Tracking Scientific Knowledge Updates in Language Models, COLM 2026β17Jun 28, 2025Updated last year
- QLoRA for Masked Language Modelingβ23Sep 11, 2023Updated 3 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.
- BrowseComp-Plus: A More Fair and Transparent Evaluation Benchmark of Deep-Research Agent (ACL 2026 Main)β387May 28, 2026Updated 5 months ago
- 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
- train with kittens!β67Oct 25, 2024Updated 2 years ago
- Repository for Multililngual Generation, RAG evaluations, and surrogate judge training for Arena RAG leaderboard (NAACL'25)β11Apr 10, 2025Updated last year
- High-Performance Engine for Multi-Vector Searchβ284Sep 10, 2026Updated last month
- β21Nov 4, 2025Updated last year
- code for training & evaluating Contextual Document Embedding modelsβ206May 14, 2025Updated last year
- GraphDancer: Training LLMs to Explore and Reason over Graphs via Curriculum Reinforcement Learningβ21Oct 18, 2026Updated 2 weeks ago
- β11Feb 9, 2024Updated 2 years ago
- Managed Database hosting by DigitalOcean β’ AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- LOFT: A 1 Million+ Token Long-Context Benchmarkβ239Apr 13, 2026Updated 6 months ago
- [EMNLP 2023] Enabling Large Language Models to Generate Text with Citations. Paper: https://arxiv.org/abs/2305.14627β531Oct 9, 2024Updated 2 years ago
- The original implementation of Min et al. "Nonparametric Masked Language Modeling" (paper https//arxiv.org/abs/2212.01349)β159Jan 6, 2023Updated 3 years ago
- Layer-Condensed KV cache w/ 10 times larger batch size, fewer params and less computation. Dramatic speed up with better task performanceβ¦β156Apr 7, 2025Updated last year
- β11May 24, 2024Updated 2 years ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Trainingβ222Aug 19, 2024Updated 2 years ago
- Positional Skip-wise Training for Efficient Context Window Extension of LLMs to Extremely Length (ICLR 2024)β209May 20, 2024Updated 2 years ago