state-spaces / s4Links
Structured state space sequence models
☆2,708Updated last year
Alternatives and similar repositories for s4
Users that are interested in s4 are comparing it to the libraries listed below
Sorting:
- A simple and efficient Mamba implementation in pure PyTorch and MLX.☆1,314Updated 8 months ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆795Updated last year
- Vector (and Scalar) Quantization, in Pytorch☆3,499Updated last week
- Simple, minimal implementation of the Mamba SSM in one file of PyTorch.☆2,849Updated last year
- Pytorch library for fast transformer implementations☆1,727Updated 2 years ago
- Implementation of https://srush.github.io/annotated-s4☆500Updated 2 months ago
- maximal update parametrization (µP)☆1,590Updated last year
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆741Updated last month
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,548Updated 2 weeks ago
- An implementation of "Retentive Network: A Successor to Transformer for Large Language Models"☆1,201Updated last year
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,170Updated 2 years ago
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,147Updated 3 years ago
- Foundation Architecture for (M)LLMs☆3,110Updated last year
- Reformer, the efficient Transformer, in Pytorch☆2,177Updated 2 years ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,157Updated 9 months ago
- Long Range Arena for Benchmarking Efficient Transformers☆762Updated last year
- Hopfield Networks is All You Need☆1,841Updated 2 years ago
- Schedule-Free Optimization in PyTorch☆2,202Updated 3 months ago
- TorchCFM: a Conditional Flow Matching library☆1,943Updated last month
- 🚀 Efficient implementations of state-of-the-art linear attention models☆3,066Updated last week
- Official repository of the xLSTM.☆1,970Updated last week
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆3,283Updated 3 months ago
- Official PyTorch implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States☆1,243Updated last year
- A Pytorch implementation of Sparsely-Gated Mixture of Experts, for massively increasing the parameter count of language models☆796Updated last year
- Collection of papers on state-space models☆596Updated 4 months ago
- ☆298Updated 7 months ago
- Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)☆1,942Updated last year
- ☆783Updated 2 months ago
- The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”☆967Updated last year
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆803Updated 2 years ago