shawntan / SUT
Repository for Sparse Universal Transformers
☆17Updated last year
Alternatives and similar repositories for SUT:
Users that are interested in SUT are comparing it to the libraries listed below
- The Energy Transformer block, in JAX☆56Updated last year
- ☆44Updated last year
- ☆22Updated 3 years ago
- Stick-breaking attention☆44Updated last month
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated last year
- Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"☆14Updated last month
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆35Updated last year
- ☆52Updated 5 months ago
- ☆28Updated 4 months ago
- ☆21Updated 5 months ago
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆36Updated last year
- Experiments on the impact of depth in transformers and SSMs.☆23Updated 3 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆25Updated 10 months ago
- Xmixers: A collection of SOTA efficient token/channel mixers☆11Updated 3 months ago
- ☆30Updated last year
- ☆47Updated last year
- ☆33Updated last year
- Official code for the paper "Attention as a Hypernetwork"☆24Updated 8 months ago
- ☆30Updated 4 months ago
- ☆18Updated 4 months ago
- ☆31Updated 10 months ago
- ☆51Updated 9 months ago
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆58Updated last year
- Yet another random morning idea to be quickly tried and architecture shared if it works; to allow the transformer to pause for any amount…☆53Updated last year
- ☆13Updated this week
- Offcial Repo of Paper "Eliminating Position Bias of Language Models: A Mechanistic Approach""☆11Updated 6 months ago
- Self-Supervised Alignment with Mutual Information☆15Updated 9 months ago
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆34Updated 2 weeks ago
- GoldFinch and other hybrid transformer components☆44Updated 7 months ago