fangyuan-ksgk / selective-attention-transformerLinks
Unofficial Implementation of Selective Attention Transformer
☆17Updated 11 months ago
Alternatives and similar repositories for selective-attention-transformer
Users that are interested in selective-attention-transformer are comparing it to the libraries listed below
Sorting:
- ☆85Updated last year
- ☆34Updated 6 months ago
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆101Updated last month
- ☆19Updated 6 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆84Updated 11 months ago
- ☆33Updated 9 months ago
- Remasking Discrete Diffusion Models with Inference-Time Scaling☆45Updated 7 months ago
- ☆20Updated last year
- Official PyTorch Implementation for Vision-Language Models Create Cross-Modal Task Representations, ICML 2025☆31Updated 5 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆54Updated last year
- The official github repo for "Diffusion Language Models are Super Data Learners".☆129Updated this week
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆131Updated 2 weeks ago
- The official implementation for [NeurIPS2025 Oral] Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink…☆83Updated 3 weeks ago
- ☆13Updated 8 months ago
- 📄Small Batch Size Training for Language Models☆62Updated last week
- DeciMamba: Exploring the Length Extrapolation Potential of Mamba (ICLR 2025)☆31Updated 6 months ago
- Official Pytorch Implementation of "The Curse of Depth in Large Language Models" by Wenfang Sun, Xinyuan Song, Pengxiang Li, Lu Yin,Yefen…☆58Updated 2 months ago
- Official code for the paper "Attention as a Hypernetwork"☆43Updated last year
- [NeurIPS '25] Multi-Token Prediction Needs Registers☆20Updated last month
- ☆33Updated 11 months ago
- The this is the official implementation of "DAPE: Data-Adaptive Positional Encoding for Length Extrapolation"☆39Updated 11 months ago
- Universal Neurons in GPT2 Language Models☆30Updated last year
- Official repository of "LiNeS: Post-training Layer Scaling Prevents Forgetting and Enhances Model Merging"☆30Updated 11 months ago
- Stick-breaking attention☆60Updated 3 months ago
- LongLLaDA: Unlocking Long Context Capabilities in Diffusion LLMs☆39Updated 2 months ago
- Multi-Layer Sparse Autoencoders (ICLR 2025)☆25Updated 7 months ago
- Code for reproducing our paper "Not All Language Model Features Are Linear"☆81Updated 10 months ago
- [ICLR 2025] Monet: Mixture of Monosemantic Experts for Transformers☆73Updated 3 months ago
- ☆58Updated last year
- [ICLR 2025] Official Pytorch Implementation of "Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LN" by Pengxia…☆26Updated 2 months ago