Scalable and Stable Parallelization of Nonlinear RNNS
☆33Jun 28, 2026Updated 4 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
- Very minimal implementation of DEER☆14Jul 6, 2026Updated 3 months ago
- ☆34Oct 22, 2024Updated 2 years ago
- Implementation of the "Online learning of long-range dependencies" paper, NeurIPS 2023☆21Nov 4, 2024Updated last year
- speed-running solving robot manipulation tasks☆24Oct 31, 2024Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆325Jan 8, 2025Updated last year
- FeedForward Propagation Through Time on Spiking Neural Network (SNNs)☆61Jan 2, 2023Updated 3 years ago
- ☆217Sep 11, 2026Updated last month
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 11 months ago
- ☆72Sep 11, 2026Updated last month
- ☆16Sep 24, 2026Updated last month
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 3 years ago
- Official implementation of the paper "Arbitrarily Scalable Environment Generators via Neural Cellular Automata", published in NeurIPS 202…☆13Oct 31, 2023Updated 2 years ago
- Streaming GPU Singular Value Decomposition (SVD) and Dynamic Mode Decomposition (DMD).☆17Jun 8, 2023Updated 3 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Official code repository for the publication "Credit Assignment in Neural Networks through Deep Feedback Control".☆11Oct 31, 2024Updated last year
- Pretraining Recurrent Networks without Recurrence☆77Jun 7, 2026Updated 4 months ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆65Sep 3, 2025Updated last year
- Repository to reproduce the results of the paper "Holomorphic Equilibrium Propagation Computes Exact Gradients Through Finite Size Oscill…☆11Oct 20, 2024Updated 2 years ago
- Deep memory and sequence models in JAX☆36Sep 25, 2026Updated last month
- ☆14Oct 11, 2022Updated 4 years ago
- Accelerated First Order Parallel Associative Scan☆198Jan 7, 2026Updated 9 months ago
- ☆18Oct 15, 2021Updated 5 years ago
- ☆15Mar 20, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134☆23May 1, 2022Updated 4 years ago
- The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…☆24Dec 30, 2022Updated 3 years ago
- Code for "Structured Linear CDEs: Maximally Expressive and Parallel-in-Time Sequence Models" (NeurIPS 2025, Spotlight)☆19Feb 17, 2026Updated 8 months ago
- Attention Kernels for Symmetric Power Transformers☆131Sep 25, 2025Updated last year
- ☆15Mar 8, 2026Updated 7 months ago
- Subset-Norm and Subset-Momentum. This repo is built on top of https://github.com/jiaweizzhao/GaLore.☆20Jul 9, 2025Updated last year
- Code and checkpoints for our paper "STAGE: Stemmed Accompaniment Generation through Prefix-Based Conditioning"☆12Sep 24, 2025Updated last year
- This is the official repo of "Scaling Recurrent Neural Networks to a Billion Parameters with Zero-Order Optimization"☆14Jan 8, 2026Updated 9 months ago
- MonarchAttention: Zero-Shot Conversion to Fast, Hardware-Aware Structured Attention (NeurIPS'25 Spotlight)☆27Sep 25, 2026Updated last month
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆25Jun 6, 2024Updated 2 years ago
- Unofficial PyTorch Implementation of "Were RNNs All We Needed?"☆17Mar 20, 2025Updated last year
- NanoGPT speedrun in JAX. Originally at https://nor-git.pages.dev/modded-nanogpt-jax/☆17Aug 28, 2025Updated last year
- ☆22Sep 29, 2024Updated 2 years ago
- Matrix-free linear algebra in JAX.☆183Oct 22, 2026Updated last week
- ☆11Jun 25, 2026Updated 4 months ago
- Efficient PScan implementation in PyTorch☆17Jan 2, 2024Updated 2 years ago