shawntan / stickbreaking-attention
Stick-breaking attention
β33Updated this week
Related projects β
Alternatives and complementary repositories for stickbreaking-attention
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"β24Updated 6 months ago
- [NeurIPS-2024] π Scaling Laws with Vocabulary: Larger Models Deserve Larger Vocabularies https://arxiv.org/abs/2407.13623β67Updated last month
- Long Context Extension and Generalization in LLMsβ39Updated last month
- β44Updated 4 months ago
- Here we will test various linear attention designs.β56Updated 6 months ago
- β24Updated 8 months ago
- DiffuGPT and DiffuLLaMA: Scaling Diffusion Language Models via Adaptation from Autoregressive Modelsβ56Updated 2 weeks ago
- β45Updated 9 months ago
- β65Updated 8 months ago
- The this is the official implementation of "DAPE: Data-Adaptive Positional Encoding for Length Extrapolation"β30Updated last month
- β50Updated 5 months ago
- Code for paper "Diffusion Language Models Can Perform Many Tasks with Scaling and Instruction-Finetuning"β63Updated 9 months ago
- Fast and memory-efficient exact attentionβ26Updated this week
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswalβ¦β44Updated last year
- A Closer Look into Mixture-of-Experts in Large Language Modelsβ39Updated 3 months ago
- Homepage for ProLong (Princeton long-context language models) and paper "How to Train Long-Context Language Models (Effectively)"β113Updated last week
- Official PyTorch Implementation of the Longhorn Deep State Space Modelβ40Updated 2 months ago
- β50Updated last week
- β25Updated 3 weeks ago
- Official github repo for the paper "Compression Represents Intelligence Linearly" [COLM 2024]β127Updated last month
- β15Updated last week
- β97Updated 8 months ago
- Official implementation of Bootstrapping Language Models via DPO Implicit Rewardsβ39Updated 3 months ago
- Simple and efficient pytorch-native transformer training and inference (batched)β61Updated 7 months ago
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"β28Updated 7 months ago
- [ACL 2024] Self-Training with Direct Preference Optimization Improves Chain-of-Thought Reasoningβ30Updated 3 months ago
- β34Updated 3 months ago
- ICML 2024 - Official Repository for EXO: Towards Efficient Exact Optimization of Language Model Alignmentβ46Updated 4 months ago
- Sparse Backpropagation for Mixture-of-Expert Trainingβ22Updated 4 months ago
- β61Updated 2 months ago