0x7o / RETRO-transformer
Easy-to-use Retrieval-Enhanced Transformer implementation
☆10Updated 2 years ago
Alternatives and similar repositories for RETRO-transformer:
Users that are interested in RETRO-transformer are comparing it to the libraries listed below
- Official repo to On the Generalization Ability of Retrieval-Enhanced Transformers☆38Updated 10 months ago
- Layer-Condensed KV cache w/ 10 times larger batch size, fewer params and less computation. Dramatic speed up with better task performance…☆148Updated 2 weeks ago
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆59Updated 6 months ago
- Official implementation for 'Extending LLMs’ Context Window with 100 Samples'☆76Updated last year
- Cascade Speculative Drafting☆29Updated last year
- ☆38Updated last year
- Implementation of the paper: "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention" from Google in pyTO…☆55Updated this week
- ☆125Updated last year
- FollowIR: Evaluating and Teaching Information Retrieval Models to Follow Instructions☆42Updated 9 months ago
- [EMNLP 2023 Industry Track] A simple prompting approach that enables the LLMs to run inference in batches.☆73Updated last year
- Retrieval as Attention☆83Updated 2 years ago
- Code repository for the c-BTM paper☆106Updated last year
- Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Fl…☆72Updated 8 months ago
- Code and Data for "Long-context LLMs Struggle with Long In-context Learning" [TMLR2025]☆105Updated 2 months ago
- minimal pytorch implementation of bm25 (with sparse tensors)☆100Updated last year
- Implementation of Speculative Sampling as described in "Accelerating Large Language Model Decoding with Speculative Sampling" by Deepmind☆93Updated last year
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆59Updated 2 months ago
- Code for Zero-Shot Tokenizer Transfer☆127Updated 3 months ago
- Codes for our paper "Speculative Decoding: Exploiting Speculative Execution for Accelerating Seq2seq Generation" (EMNLP 2023 Findings)☆39Updated last year
- [EMNLP 2024] A Retrieval Benchmark for Scientific Literature Search☆83Updated 4 months ago
- ☆50Updated 5 months ago
- REST: Retrieval-Based Speculative Decoding, NAACL 2024☆199Updated 4 months ago
- Spherical Merge Pytorch/HF format Language Models with minimal feature loss.☆120Updated last year
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆40Updated last year
- ☆33Updated 10 months ago
- Repo for ICML23 "Why do Nearest Neighbor Language Models Work?"☆56Updated 2 years ago
- DSIR large-scale data selection framework for language model training☆246Updated last year
- This repository contains the joint use of CPO and SimPO method for better reference-free preference learning methods.☆53Updated 8 months ago
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…☆128Updated 8 months ago
- ☆72Updated 11 months ago