shawntan / stickbreaking-attention
Stick-breaking attention
☆34Updated last week
Related projects ⓘ
Alternatives and complementary repositories for stickbreaking-attention
- ☆45Updated 9 months ago
- ☆24Updated 8 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆24Updated 7 months ago
- ☆45Updated 4 months ago
- ☆50Updated 6 months ago
- Here we will test various linear attention designs.☆56Updated 6 months ago
- Long Context Extension and Generalization in LLMs☆39Updated 2 months ago
- [NeurIPS-2024] 📈 Scaling Laws with Vocabulary: Larger Models Deserve Larger Vocabularies https://arxiv.org/abs/2407.13623☆69Updated last month
- The this is the official implementation of "DAPE: Data-Adaptive Positional Encoding for Length Extrapolation"☆32Updated last month
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…☆44Updated last year
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆40Updated 3 months ago
- Fast and memory-efficient exact attention☆27Updated last week
- Official Implementation Of The Paper: `DeciMamba: Exploring the Length Extrapolation Potential of Mamba'☆20Updated 3 months ago
- ☆69Updated 8 months ago
- DiffuGPT and DiffuLLaMA: Scaling Diffusion Language Models via Adaptation from Autoregressive Models☆57Updated 3 weeks ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆61Updated 6 months ago
- Code for paper "Diffusion Language Models Can Perform Many Tasks with Scaling and Instruction-Finetuning"☆63Updated 9 months ago
- Using FlexAttention to compute attention with different masking patterns☆40Updated last month
- ☆53Updated 3 weeks ago
- ☆17Updated this week
- ☆44Updated last year
- ICML 2024 - Official Repository for EXO: Towards Efficient Exact Optimization of Language Model Alignment☆46Updated 5 months ago
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…☆78Updated 2 months ago
- ☆98Updated 8 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆49Updated 3 months ago
- ☆75Updated last year
- Homepage for ProLong (Princeton long-context language models) and paper "How to Train Long-Context Language Models (Effectively)"☆118Updated 3 weeks ago
- [ACL 2024] Self-Training with Direct Preference Optimization Improves Chain-of-Thought Reasoning☆30Updated 3 months ago
- ☆35Updated 7 months ago
- Code for the paper: Why Transformers Need Adam: A Hessian Perspective☆42Updated 6 months ago