☆320Jan 8, 2025Updated last year
Alternatives and similar repositories for S5
Users that are interested in S5 are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆87Apr 26, 2024Updated 2 years ago
- Structured state space sequence models☆2,895Jul 17, 2024Updated last year
- Implementation of https://srush.github.io/annotated-s4☆517Jun 20, 2025Updated 11 months ago
- ☆165Jan 24, 2023Updated 3 years ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆62Sep 3, 2025Updated 8 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Sequence Modeling with Structured State Spaces☆67Aug 2, 2022Updated 3 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆114Dec 5, 2023Updated 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
- Liquid Structural State-Space Models☆396Feb 1, 2024Updated 2 years ago
- Sequence modeling with Mega.☆303Jan 28, 2023Updated 3 years ago
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆77Apr 22, 2025Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆84May 17, 2024Updated 2 years ago
- Convolutions for Sequence Modeling☆912Jun 13, 2024Updated last year
- Code for SpaceTime 🌌⏱️. Proposed in Effectively Modeling Time Series with Simple Discrete State Spaces, ICLR 2023.☆181Mar 17, 2023Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Scalable and Stable Parallelization of Nonlinear RNNS☆31Mar 6, 2026Updated 2 months ago
- ☆36Nov 22, 2024Updated last year
- Implementation of the "Online learning of long-range dependencies" paper, NeurIPS 2023☆21Nov 4, 2024Updated last year
- POPGym Library in JAX☆14Apr 15, 2024Updated 2 years ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆68Apr 24, 2024Updated 2 years ago
- ☆35Apr 12, 2024Updated 2 years ago
- ☆20Apr 17, 2023Updated 3 years ago
- Experiments on the impact of depth in transformers and SSMs.☆40Oct 23, 2025Updated 6 months ago
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆22Jul 14, 2024Updated last year
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Parallelizing non-linear sequential models over the sequence length☆57Jun 23, 2025Updated 10 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆57Aug 20, 2024Updated last year
- Accelerated First Order Parallel Associative Scan☆197Jan 7, 2026Updated 4 months ago
- Implementations of various linear RNN layers using pytorch and triton☆55Aug 4, 2023Updated 2 years ago
- ☆11Oct 11, 2023Updated 2 years ago
- Collection of papers on state-space models☆621Nov 4, 2025Updated 6 months ago
- Annotated version of the Mamba paper☆501Feb 27, 2024Updated 2 years ago
- ☆29Jul 9, 2024Updated last year
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆98Dec 5, 2024Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆16Jan 7, 2025Updated last year
- ☆45Apr 30, 2018Updated 8 years ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆101Feb 25, 2023Updated 3 years ago
- A Python package for probabilistic state space modeling with JAX☆964Jan 6, 2026Updated 4 months ago
- PyTorch implementation of the NCDSSM models presented in the ICML '23 paper "Neural Continuous-Discrete State Space Models for Irregularl…☆27Jul 9, 2023Updated 2 years ago
- ☆201May 27, 2024Updated last year
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆57Dec 4, 2024Updated last year