kyegomez / Hedgehog
Implementation of the model "Hedgehog" from the paper: "The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicry"
☆13Updated last year
Alternatives and similar repositories for Hedgehog:
Users that are interested in Hedgehog are comparing it to the libraries listed below
- Official code for the paper "Attention as a Hypernetwork"☆25Updated 9 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆53Updated 7 months ago
- ☆30Updated 2 months ago
- ☆18Updated 8 months ago
- Official implementation of ECCV24 paper: POA☆24Updated 7 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆26Updated 11 months ago
- Collect papers about Mamba (a selective state space model).☆14Updated 7 months ago
- Official Pytorch Implementation of "Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LN" by Pengxiang Li, Lu Yi…☆19Updated 3 months ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆17Updated last year
- Lottery Ticket Adaptation☆38Updated 4 months ago
- DeciMamba: Exploring the Length Extrapolation Potential of Mamba (ICLR 2025)☆23Updated 8 months ago
- Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classification☆11Updated last year
- Official implementation of RMoE (Layerwise Recurrent Router for Mixture-of-Experts)☆18Updated 7 months ago
- Code for "Merging Text Transformers from Different Initializations"☆19Updated last month
- The official repository for our paper "The Dual Form of Neural Networks Revisited: Connecting Test Time Predictions to Training Patterns …☆16Updated last year
- Official repository of "LiNeS: Post-training Layer Scaling Prevents Forgetting and Enhances Model Merging"☆24Updated 4 months ago
- This is the oficial repository for "Safer-Instruct: Aligning Language Models with Automated Preference Data"☆17Updated last year
- Official repo of Progressive Data Expansion: data, code and evaluation☆28Updated last year
- ☆30Updated 9 months ago
- Official repository for ICML 2024 paper "MoRe Fine-Tuning with 10x Fewer Parameters"☆17Updated last week
- Experimental scripts for researching data adaptive learning rate scheduling.☆23Updated last year
- Implementation of the model: "Reka Core, Flash, and Edge: A Series of Powerful Multimodal Language Models" in PyTorch☆30Updated last week
- Self Reproduction Code of Paper "Reducing Transformer Key-Value Cache Size with Cross-Layer Attention (MIT CSAIL)☆12Updated 10 months ago
- Efficient Scaling laws and collaborative pretraining.☆15Updated last month
- [ICML 2024] When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Models☆30Updated 9 months ago
- This is a simple torch implementation of the high performance Multi-Query Attention☆16Updated last year
- Here we will test various linear attention designs.☆60Updated 10 months ago
- Official implementation for Sparse MetA-Tuning (SMAT)☆16Updated 8 months ago