schwartz-lab-NLP / TOVA
Token Omission Via Attention
☆121Updated last month
Related projects ⓘ
Alternatives and complementary repositories for TOVA
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆104Updated 2 months ago
- Understand and test language model architectures on synthetic tasks.☆163Updated 6 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆214Updated this week
- Parameter-Efficient Sparsity Crafting From Dense to Mixture-of-Experts for Instruction Tuning on General Tasks☆129Updated 2 months ago
- Layer-Condensed KV cache w/ 10 times larger batch size, fewer params and less computation. Dramatic speed up with better task performance…☆139Updated this week
- ☆184Updated last month
- ☆122Updated 10 months ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆181Updated last month
- ☆62Updated 3 months ago
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"☆93Updated last month
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆84Updated this week
- Griffin MQA + Hawk Linear RNN Hybrid☆85Updated 6 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆95Updated 6 months ago
- Triton Implementation of HyperAttention Algorithm☆46Updated 11 months ago
- ☆74Updated 11 months ago
- A repository for research on medium sized language models.☆74Updated 6 months ago
- Simple and efficient pytorch-native transformer training and inference (batched)☆61Updated 7 months ago
- Some preliminary explorations of Mamba's context scaling.☆191Updated 9 months ago
- ☆55Updated last month
- The official repo for "LLoCo: Learning Long Contexts Offline"☆113Updated 5 months ago
- Normalized Transformer (nGPT)☆87Updated this week
- Language models scale reliably with over-training and on downstream tasks☆94Updated 7 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆112Updated 3 months ago
- This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors" in ICML 2024.☆82Updated 4 months ago
- Plug in & Play Pytorch Implementation of the paper: "Evolutionary Optimization of Model Merging Recipes" by Sakana AI☆26Updated last week
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…☆87Updated 3 months ago
- Positional Skip-wise Training for Efficient Context Window Extension of LLMs to Extremely Length (ICLR 2024)☆199Updated 6 months ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (Official Code)☆135Updated last month
- Replicating O1 inference-time scaling laws☆49Updated last month
- ☆50Updated 6 months ago