DonRL10 / RetNetLinks
an implementation of paper"Retentive Network: A Successor to Transformer for Large Language Models" https://arxiv.org/pdf/2307.08621.pdf
☆12Updated 2 years ago
Alternatives and similar repositories for RetNet
Users that are interested in RetNet are comparing it to the libraries listed below
Sorting:
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆65Updated last year
 - ☆56Updated last year
 - ☆67Updated last year
 - ☆14Updated 2 years ago
 - Implementation of GateLoop Transformer in Pytorch and Jax☆90Updated last year
 - Contextual Position Encoding but with some custom CUDA Kernels https://arxiv.org/abs/2405.18719☆22Updated last year
 - Code for the paper "The Impact of Positional Encoding on Length Generalization in Transformers", NeurIPS 2023☆136Updated last year
 - Griffin MQA + Hawk Linear RNN Hybrid☆89Updated last year
 - Randomized Positional Encodings Boost Length Generalization of Transformers☆83Updated last year
 - ☆27Updated last year
 - Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆15Updated last year
 - A (somewhat) minimal library for finetuning language models with PPO on human feedback.☆87Updated 2 years ago
 - ☆12Updated last year
 - Implementation of the paper: "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention" from Google in pyTO…☆56Updated 2 weeks ago
 - ☆48Updated last year
 - RWKV, in easy to read code☆72Updated 7 months ago
 - ☆33Updated last week
 - ☆19Updated 3 years ago
 - Pytorch implementation of "Block Recurrent Transformers" (Hutchins & Schlag et al., 2022)☆84Updated 3 years ago
 - Efficient PScan implementation in PyTorch☆16Updated last year
 - Implementation of "LM-Infinite: Simple On-the-Fly Length Generalization for Large Language Models"☆39Updated 11 months ago
 - Efficient Infinite Context Transformers with Infini-attention Pytorch Implementation + QwenMoE Implementation + Training Script + 1M cont…☆84Updated last year
 - Code for ICML 25 paper "Metadata Conditioning Accelerates Language Model Pre-training (MeCo)"☆46Updated 4 months ago
 - ☆26Updated 3 months ago
 - A large-scale RWKV v7(World, PRWKV, Hybrid-RWKV) inference. Capable of inference by combining multiple states(Pseudo MoE). Easy to deploy…☆45Updated last week
 - Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆129Updated this week
 - GoldFinch and other hybrid transformer components☆45Updated last year
 - ☆31Updated last year
 - Efficient Transformers with Dynamic Token Pooling☆64Updated 2 years ago
 - Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"☆100Updated last year