Scalable and Stable Parallelization of Nonlinear RNNS
☆33Jun 28, 2026Updated 2 months ago
Alternatives and similar repositories for elk
Users that are interested in elk are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Parallelizing non-linear sequential models over the sequence length☆56Jun 23, 2025Updated last year
- Training Recurrent Neural Networks via Forward Propagation Through Time☆42Jun 11, 2021Updated 5 years ago
- Implementation of the "Online learning of long-range dependencies" paper, NeurIPS 2023☆21Nov 4, 2024Updated last year
- HGRN2: Gated Linear RNNs with State Expansion☆59Aug 20, 2024Updated 2 years ago
- ☆204Oct 31, 2025Updated 9 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 9 months ago
- speed-running solving robot manipulation tasks☆24Oct 31, 2024Updated last year
- FeedForward Propagation Through Time on Spiking Neural Network (SNNs)☆60Jan 2, 2023Updated 3 years ago
- ☆21Jan 4, 2023Updated 3 years ago
- ☆12May 16, 2024Updated 2 years ago
- Code for "Structured Linear CDEs: Maximally Expressive and Parallel-in-Time Sequence Models" (NeurIPS 2025, Spotlight)☆19Feb 17, 2026Updated 6 months ago
- ☆16Updated this week
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 2 years ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆63Sep 3, 2025Updated 11 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆25Jun 6, 2024Updated 2 years ago
- Deep memory and sequence models in JAX☆32Jun 8, 2026Updated 2 months ago
- ☆324Jan 8, 2025Updated last year
- ☆15Aug 21, 2026Updated last week
- Code for "Learning Unitary Operators with Help From u(n)", AAAI-17. (https://arxiv.org/abs/1607.04903)☆17Jan 10, 2017Updated 9 years ago
- Repository to reproduce the results of the paper "Holomorphic Equilibrium Propagation Computes Exact Gradients Through Finite Size Oscill…☆11Oct 20, 2024Updated last year
- Comparison between GFlowNets & Maximum Entropy RL☆19Feb 19, 2024Updated 2 years ago
- GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134☆23May 1, 2022Updated 4 years ago
- Implementation of Hyena Hierarchy in JAX☆10Apr 30, 2023Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ☆15Mar 20, 2025Updated last year
- ☆18Oct 15, 2021Updated 4 years ago
- Experiments on the impact of depth in transformers and SSMs.☆46Oct 23, 2025Updated 10 months ago
- A State-Space Model with Rational Transfer Function Representation.☆84May 17, 2024Updated 2 years ago
- FlashRNN - Fast RNN Kernels with I/O Awareness☆188Jul 29, 2026Updated last month
- [ICCV 2025] Code for "Flow Stochastic Segmentation Networks"☆18Jun 9, 2026Updated 2 months ago
- Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)☆17Oct 11, 2021Updated 4 years ago
- MonarchAttention: Zero-Shot Conversion to Fast, Hardware-Aware Structured Attention (NeurIPS'25 Spotlight)☆26Feb 22, 2026Updated 6 months ago
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- DeltaProduct is a new linear recurrent neural network architecture that uses products of generalized Householder matrices as state-transi…☆17Oct 13, 2025Updated 10 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Oct 11, 2023Updated 2 years ago
- Subset-Norm and Subset-Momentum. This repo is built on top of https://github.com/jiaweizzhao/GaLore.☆19Jul 9, 2025Updated last year
- ☆35Apr 12, 2024Updated 2 years ago
- AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)☆24Oct 15, 2024Updated last year
- ☆16Aug 4, 2026Updated 3 weeks ago
- Code for lin-RFM used for sparse recovery tasks☆17Mar 13, 2025Updated last year