TobiasNorlund / retro
Official repo to On the Generalization Ability of Retrieval-Enhanced Transformers
☆38Updated 11 months ago
Alternatives and similar repositories for retro:
Users that are interested in retro are comparing it to the libraries listed below
- ☆103Updated 11 months ago
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆40Updated last year
- ☆38Updated last year
- ☆143Updated last year
- ☆34Updated last year
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆91Updated last year
- Repo for ICML23 "Why do Nearest Neighbor Language Models Work?"☆56Updated 2 years ago
- The official repo for "LLoCo: Learning Long Contexts Offline"☆116Updated 10 months ago
- ☆72Updated last year
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆60Updated 3 months ago
- ☆72Updated last year
- Code repository for the c-BTM paper☆106Updated last year
- some common Huggingface transformers in maximal update parametrization (µP)☆80Updated 3 years ago
- Easy-to-use Retrieval-Enhanced Transformer implementation☆10Updated 2 years ago
- Simple and efficient pytorch-native transformer training and inference (batched)☆75Updated last year
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆209Updated 8 months ago
- ☆81Updated last year
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆59Updated 7 months ago
- Retrieval as Attention☆83Updated 2 years ago
- ☆104Updated 8 months ago
- Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"☆116Updated last year
- CUDA and Triton implementations of Flash Attention with SoftmaxN.☆70Updated 11 months ago
- Multipack distributed sampler for fast padding-free training of LLMs☆188Updated 9 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆212Updated 5 months ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆65Updated 9 months ago
- Randomized Positional Encodings Boost Length Generalization of Transformers☆81Updated last year
- ☆49Updated last year
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…☆131Updated 9 months ago
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆65Updated 5 months ago
- ☆50Updated 6 months ago