state-spaces / s4
Structured state space sequence models
☆2,589Updated 8 months ago
Alternatives and similar repositories for s4:
Users that are interested in s4 are comparing it to the libraries listed below
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,106Updated 3 months ago
- Simple, minimal implementation of the Mamba SSM in one file of PyTorch.☆2,753Updated last year
- Implementation of https://srush.github.io/annotated-s4☆486Updated 2 years ago
- A simple and efficient Mamba implementation in pure PyTorch and MLX.☆1,171Updated 3 months ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆651Updated 3 months ago
- Pytorch library for fast transformer implementations☆1,687Updated 2 years ago
- Mamba SSM architecture☆14,343Updated 2 months ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆751Updated 10 months ago
- Long Range Arena for Benchmarking Efficient Transformers☆748Updated last year
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,116Updated 3 years ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,160Updated this week
- Foundation Architecture for (M)LLMs☆3,062Updated 11 months ago
- An implementation of "Retentive Network: A Successor to Transformer for Large Language Models"☆1,177Updated last year
- maximal update parametrization (µP)☆1,481Updated 8 months ago
- Vector (and Scalar) Quantization, in Pytorch☆3,043Updated this week
- ☆771Updated 2 months ago
- Collection of papers on state-space models☆584Updated 3 weeks ago
- The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”☆955Updated last year
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,134Updated last year
- Awesome Papers related to Mamba.☆1,333Updated 5 months ago
- ☆175Updated 9 months ago
- Official PyTorch implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States☆1,144Updated 8 months ago
- Reformer, the efficient Transformer, in Pytorch☆2,155Updated last year
- ☆286Updated 2 months ago
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,848Updated last year
- Schedule-Free Optimization in PyTorch☆2,118Updated this week
- Convolutions for Sequence Modeling☆878Updated 9 months ago
- Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)☆1,888Updated 7 months ago
- PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡☆4,522Updated 7 months ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆2,232Updated 2 months ago