☆316Jan 8, 2025Updated last year
Alternatives and similar repositories for S5
Users that are interested in S5 are comparing it to the libraries listed below
Sorting:
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆82Apr 26, 2024Updated last year
- Structured state space sequence models☆2,854Jul 17, 2024Updated last year
- ☆164Jan 24, 2023Updated 3 years ago
- Implementation of https://srush.github.io/annotated-s4☆512Jun 20, 2025Updated 8 months ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆62Sep 3, 2025Updated 5 months ago
- Sequence Modeling with Structured State Spaces☆67Aug 2, 2022Updated 3 years ago
- Liquid Structural State-Space Models☆384Feb 1, 2024Updated 2 years ago
- The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…☆23Dec 30, 2022Updated 3 years ago
- Sequence modeling with Mega.☆303Jan 28, 2023Updated 3 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆112Dec 5, 2023Updated 2 years ago
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆74Apr 22, 2025Updated 10 months ago
- A State-Space Model with Rational Transfer Function Representation.☆83May 17, 2024Updated last year
- Convolutions for Sequence Modeling☆912Jun 13, 2024Updated last year
- ☆35Apr 12, 2024Updated last year
- ☆20Apr 17, 2023Updated 2 years ago
- ☆35Nov 22, 2024Updated last year
- Code for SpaceTime 🌌⏱️. Proposed in Effectively Modeling Time Series with Simple Discrete State Spaces, ICLR 2023.☆181Mar 17, 2023Updated 2 years ago
- Parallelizing non-linear sequential models over the sequence length☆56Jun 23, 2025Updated 8 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆56Aug 20, 2024Updated last year
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆67Apr 24, 2024Updated last year
- POPGym Library in JAX☆12Apr 15, 2024Updated last year
- ☆11Oct 11, 2023Updated 2 years ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆98Dec 5, 2024Updated last year
- ☆13Dec 15, 2025Updated 2 months ago
- ☆29Jul 9, 2024Updated last year
- AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)☆23Oct 15, 2024Updated last year
- Accelerated First Order Parallel Associative Scan☆194Jan 7, 2026Updated last month
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Jan 7, 2025Updated last year
- Implementations of various linear RNN layers using pytorch and triton☆54Aug 4, 2023Updated 2 years ago
- Experiments on the impact of depth in transformers and SSMs.☆40Oct 23, 2025Updated 4 months ago
- Scalable and Stable Parallelization of Nonlinear RNNS☆29Oct 21, 2025Updated 4 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Oct 11, 2023Updated 2 years ago
- Collection of papers on state-space models☆615Nov 4, 2025Updated 3 months ago
- ☆45Apr 30, 2018Updated 7 years ago
- ☆51Jan 28, 2024Updated 2 years ago
- Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆15May 10, 2024Updated last year
- [ICLR 2023] Official implementation of Transnormer in our ICLR 2023 paper - Toeplitz Neural Network for Sequence Modeling☆81Apr 24, 2024Updated last year
- Annotated version of the Mamba paper☆497Feb 27, 2024Updated 2 years ago
- ☆29Feb 27, 2024Updated 2 years ago