RobertCsordas / moe_attentionLinks
Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"
β102Updated last year
Alternatives and similar repositories for moe_attention
Users that are interested in moe_attention are comparing it to the libraries listed below
Sorting:
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmindβ134Updated 3 months ago
- DPO, but faster πβ46Updated last year
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)β163Updated 9 months ago
- A repository for research on medium sized language models.β77Updated last year
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"β38Updated 7 months ago
- β128Updated 2 years ago
- β91Updated last year
- Official implementation of the ICML 2024 paper RoSA (Robust Adaptation)β44Updated last year
- [ICML'24] The official implementation of βRethinking Optimization and Architecture for Tiny Language Modelsββ126Updated last year
- Parameter-Efficient Sparsity Crafting From Dense to Mixture-of-Experts for Instruction Tuning on General Tasks (EMNLP'24)β147Updated last year
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.β50Updated 2 years ago
- Token Omission Via Attentionβ128Updated last year
- β54Updated last year
- The official repo for "LLoCo: Learning Long Contexts Offline"β118Updated last year
- Implementation of the paper: "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention" from Google in pyTOβ¦β58Updated 2 weeks ago
- β203Updated last year
- β85Updated 2 months ago
- Triton Implementation of HyperAttention Algorithmβ48Updated 2 years ago
- Griffin MQA + Hawk Linear RNN Hybridβ88Updated last year
- Implementation of Infini-Transformer in Pytorchβ112Updated last year
- A byte-level decoder architecture that matches the performance of tokenized Transformers.β67Updated last year
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear atβ¦β104Updated last year
- [TMLR 2026] When Attention Collapses: How Degenerate Layers in LLMs Enable Smaller, Stronger Modelsβ121Updated 11 months ago
- PyTorch implementation of models from the Zamba2 series.β186Updated last year
- [NeurIPS 2024] Low rank memory efficient optimizer without SVDβ33Updated 7 months ago
- PB-LLM: Partially Binarized Large Language Modelsβ157Updated 2 years ago
- RWKV-7: Surpassing GPTβ104Updated last year
- Here we will test various linear attention designs.β62Updated last year
- Linear Attention Sequence Parallelism (LASP)β88Updated last year
- Unofficial Implementation of Evolutionary Model Mergingβ41Updated last year