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
- ☆45Updated last year
- ☆32Updated 7 months ago
- The Energy Transformer block, in JAX☆57Updated last year
- Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"☆14Updated 3 months ago
- ☆31Updated last year
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆37Updated last year
- ☆22Updated 3 years ago
- Experiments on the impact of depth in transformers and SSMs.☆25Updated 6 months ago
- ☆21Updated 7 months ago
- ☆52Updated 7 months ago
- ☆51Updated 11 months ago
- ☆28Updated last month
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- ☆11Updated last year
- Official repository for the paper "Exploring the Promise and Limits of Real-Time Recurrent Learning" (ICLR 2024)☆11Updated last year
- ☆32Updated last year
- 🧮 Algebraic Positional Encodings.☆12Updated 3 months ago
- ☆19Updated 2 months ago
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated last year
- ☆47Updated last year
- Stick-breaking attention☆52Updated last month
- Parallel Associative Scan for Language Models☆18Updated last year
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- Scalable and Stable Parallelization of Nonlinear RNNS☆15Updated 3 months ago
- Xmixers: A collection of SOTA efficient token/channel mixers☆11Updated 5 months ago
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [to appear at ICLR 2025]☆19Updated last month
- Code for NeurIPS 2024 paper: "Noether's razor: Learning Conserved Quantities" by Tycho F. A. van der Ouderaa, Mark van der Wilk, Pim de H…☆10Updated 6 months ago
- ☆31Updated 6 months ago
- Code for "Discovering Non-monotonic Autoregressive Orderings with Variational Inference" (paper and code updated from ICLR 2021)☆12Updated last year
- Official Code Repository for the paper "Key-value memory in the brain"☆25Updated 2 months ago