RobertCsordas / moe_attention
Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"
☆96Updated 4 months ago
Alternatives and similar repositories for moe_attention:
Users that are interested in moe_attention are comparing it to the libraries listed below
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆149Updated last month
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆36Updated last year
- My fork os allen AI's OLMo for educational purposes.☆30Updated 2 months ago
- This repo is based on https://github.com/jiaweizzhao/GaLore☆24Updated 4 months ago
- ☆74Updated last month
- PB-LLM: Partially Binarized Large Language Models☆150Updated last year
- ☆71Updated 5 months ago
- A repository for research on medium sized language models.☆76Updated 8 months ago
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆194Updated 2 weeks ago
- ☆82Updated 4 months ago
- [ICML'24] The official implementation of “Rethinking Optimization and Architecture for Tiny Language Models”☆120Updated last month
- ☆43Updated 3 months ago
- ☆125Updated last year
- Train, tune, and infer Bamba model☆83Updated 3 weeks ago
- RWKV-7: Surpassing GPT☆76Updated 2 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆117Updated 5 months ago
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…☆96Updated 5 months ago
- This is the official repository for Inheritune.☆109Updated this week
- QuIP quantization☆48Updated 10 months ago
- The official repo for "LLoCo: Learning Long Contexts Offline"☆114Updated 8 months ago
- Parameter-Efficient Sparsity Crafting From Dense to Mixture-of-Experts for Instruction Tuning on General Tasks☆140Updated 4 months ago
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.☆44Updated last year
- Unofficial Implementation of Evolutionary Model Merging☆33Updated 10 months ago
- Griffin MQA + Hawk Linear RNN Hybrid☆85Updated 9 months ago
- Implementation of the paper: "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆81Updated this week
- A byte-level decoder architecture that matches the performance of tokenized Transformers.☆65Updated 9 months ago
- ☆99Updated 11 months ago
- A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM☆50Updated 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…☆147Updated 3 weeks ago
- Some preliminary explorations of Mamba's context scaling.☆213Updated last year