Non-official implementation of "Attention as an RNN" from https://arxiv.org/pdf/2405.13956, efficient associative parallel prefix scan and recurrent version implemented.
☆27Jul 27, 2024Updated last year
Alternatives and similar repositories for Attention-as-RNN
Users that are interested in Attention-as-RNN are comparing it to the libraries listed below
Sorting:
- Official Implementation of Infinite-Resolution Integral Noise Warping for Diffusion Models [ICLR 2025]☆15Mar 15, 2025Updated last year
- A single-line modification to any (dualizer-based) optimizer that allows the optimizer to adapt to the scale of the gradients as they cha…☆19Jan 11, 2025Updated last year
- Zicx's Notebook.☆10Nov 7, 2025Updated 4 months ago
- ☆17Jul 24, 2023Updated 2 years ago
- This repository is the official implementation of Low-Rank Modular Reinforcement Learning via Muscle Synergy.☆11Oct 27, 2022Updated 3 years ago
- [ICLR 2025] This repo is the official implementation of "The Labyrinth of Links: Navigating the Associative Maze of Multi-modal LLMs".☆13Jan 25, 2025Updated last year
- ☆20Jan 4, 2026Updated 2 months ago
- ☆12Dec 15, 2022Updated 3 years ago
- [IEEE TETCI 2025] Official repository of the PETformer paper: "PETformer: Long-term Time Series Forecasting via Placeholder-enhanced Tran…☆13Dec 6, 2024Updated last year
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆25Jan 9, 2026Updated 2 months ago
- [ICLR 2025] "Understanding Constraint Inference in Safety-Critical Inverse Reinforcement Learning"☆13Nov 30, 2025Updated 3 months ago
- Implementation for paper "Link Prediction on Heterophilic Graphs via Disentangled Representation Learning"☆13Aug 26, 2022Updated 3 years ago
- A novel approach for transformer model introspection that enables saving, compressing, and manipulating internal thought states for advan…☆31Mar 3, 2026Updated 2 weeks ago
- ☆20Mar 3, 2025Updated last year
- ☆17Sep 25, 2025Updated 5 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Jun 5, 2025Updated 9 months ago
- ☆24Sep 25, 2024Updated last year
- Planning Algorithm for All-Wheel-Steering Platforms☆20Jan 19, 2026Updated 2 months ago
- Train toy models using multi-token prediction objective☆14May 8, 2024Updated last year
- High order Moment Models☆41Nov 13, 2025Updated 4 months ago
- Mariana Pro colorscheme from Sublime Text ported to Vim☆10Aug 2, 2025Updated 7 months ago
- The official code of FeRA: Frequency–Energy Constrained Routing for Effective Diffusion Adaptation Fine-Tuning☆29Dec 27, 2025Updated 2 months ago
- [NeurIPS 2024] Official implementation of NeurIPS 2024 paepr "Flow Priors for Linear Inverse Problems via Iterative Corrupted Trajectory …☆26Feb 24, 2025Updated last year
- ☆19Jul 21, 2019Updated 6 years ago
- ☆20Aug 6, 2024Updated last year
- 利用海康威视的代码进行目标检测与跟踪☆21Jan 4, 2024Updated 2 years ago
- Text-to-Speech for ROS 2☆21Dec 8, 2025Updated 3 months ago
- ☆11Nov 13, 2025Updated 4 months ago
- Parallel Prefix Sum (Scan) with CUDA.☆15Jul 17, 2020Updated 5 years ago
- Official PyTorch implementation for the paper "Interpretable Image Classification via Non-parametric Part Prototype Learning" CVPR 2025.☆27Jun 10, 2025Updated 9 months ago
- Accepted at WWW 25 Industrial Track (oral)☆17Jun 6, 2025Updated 9 months ago
- ☆31Nov 8, 2023Updated 2 years ago
- Artificial Neural-Net compression and fitting of synthetic spectral grids.☆21Nov 14, 2025Updated 4 months ago
- Official implementation for "Let Offline RL Flow: Training Conservative Agents in the Latent Space of Normalizing Flows", NeurIPS 2022, O…☆12Jan 31, 2023Updated 3 years ago
- ImMesh_PGO: An Immediate LiDAR Localization and Meshing Framework with Loop Closure☆33Nov 23, 2024Updated last year
- A Fast, Portable Deep Reinforcement Learning Library for Continuous Control☆13Jul 26, 2023Updated 2 years ago
- Official implementation of the UMDQN algorithm presented in the scientific research paper entitled "Distributional Reinforcement Learning…☆11Jun 3, 2022Updated 3 years ago
- Reference implementation of models from Nyonic Model Factory☆12May 13, 2024Updated last year
- ☆15Jun 1, 2023Updated 2 years ago