lindermanlab / elkLinks
Scalable and Stable Parallelization of Nonlinear RNNS
☆23Updated this week
Alternatives and similar repositories for elk
Users that are interested in elk are comparing it to the libraries listed below
Sorting:
- ☆58Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆81Updated last year
- ☆33Updated last year
- The Energy Transformer block, in JAX☆58Updated last year
- Experiments on the impact of depth in transformers and SSMs.☆34Updated 11 months ago
- ☆33Updated 10 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆89Updated last year
- ☆120Updated 3 months ago
- ☆34Updated last year
- supporting pytorch FSDP for optimizers☆84Updated 10 months ago
- ☆53Updated last year
- Implementation of GateLoop Transformer in Pytorch and Jax☆90Updated last year
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Updated last year
- nanoGPT using Equinox☆13Updated 2 years ago
- Parallelizing non-linear sequential models over the sequence length☆54Updated 3 months ago
- Accelerated First Order Parallel Associative Scan☆190Updated last year
- ☆39Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆129Updated 9 months ago
- ☆53Updated last year
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14Updated 4 months ago
- ☆52Updated last year
- Code for the paper "Function-Space Learning Rates"☆23Updated 4 months ago
- LoRA for arbitrary JAX models and functions☆142Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated last year
- An annotated implementation of the Hyena Hierarchy paper☆34Updated 2 years ago
- ☆17Updated last year
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆51Updated last year
- ☆67Updated 10 months ago
- Supporting code for the blog post on modular manifolds.☆71Updated last week
- ☆215Updated 10 months ago