test-time-training / ttt-lm-kernels
Inference Speed Benchmark for Learning to (Learn at Test Time): RNNs with Expressive Hidden States
β59Updated 7 months ago
Alternatives and similar repositories for ttt-lm-kernels:
Users that are interested in ttt-lm-kernels are comparing it to the libraries listed below
- π₯ A minimal training framework for scaling FLA modelsβ59Updated this week
- HGRN2: Gated Linear RNNs with State Expansionβ52Updated 6 months ago
- Linear Attention Sequence Parallelism (LASP)β77Updated 8 months ago
- β99Updated 11 months ago
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Ruleβ131Updated last week
- A repository for DenseSSMsβ86Updated 10 months ago
- Here we will test various linear attention designs.β58Updated 9 months ago
- [ICML 2024] When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Modelsβ28Updated 8 months ago
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"β96Updated 4 months ago
- Code for paper "Patch-Level Training for Large Language Models"β80Updated 3 months ago
- Implementation of Griffin from the paper: "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"β51Updated 3 weeks ago
- β24Updated 4 months ago
- [EVA ICLR'23; LARA ICML'22] Efficient attention mechanisms via control variates, random features, and importance samplingβ80Updated last year
- Stick-breaking attentionβ43Updated last month
- Official Implementation Of The Paper: `DeciMamba: Exploring the Length Extrapolation Potential of Mamba'β23Updated 6 months ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Seβ¦β62Updated 9 months ago
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Modeβ¦β96Updated 5 months ago
- [EMNLP 2022] Official implementation of Transnormer in our EMNLP 2022 paper - The Devil in Linear Transformerβ59Updated last year
- β187Updated last year
- [ICML 2024 Oral] This project is the official implementation of our Accurate LoRA-Finetuning Quantization of LLMs via Information Retentiβ¦β60Updated 10 months ago
- An unofficial implementation of "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"β35Updated 8 months ago
- [ICLR 2025] Official Code Release for Explaining Modern Gated-Linear RNNs via a Unified Implicit Attention Formulationβ37Updated 8 months ago
- Activation-aware Singular Value Decomposition for Compressing Large Language Modelsβ56Updated 3 months ago
- Triton implement of bi-directional (non-causal) linear attentionβ41Updated 2 weeks ago
- Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for LLMsβ78Updated 2 months ago
- Fast and memory-efficient exact attentionβ58Updated this week
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswalβ¦β48Updated last year
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.β45Updated last year
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear atβ¦β99Updated 8 months ago
- Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.β57Updated 2 months ago