Paper Reading AI Learner

$texttt{lrnnx}$: A library for Linear RNNs

2026-02-09 15:48:48
Karan Bania, Soham Kalburgi, Manit Tanwar, Dhruthi, Aditya Nagarsekar, Harshvardhan Mestha, Naman Chibber, Raj Deshmukh, Anish Sathyanarayanan, Aarush Rathore, Pratham Chheda

Abstract

Linear recurrent neural networks (LRNNs) provide a structured approach to sequence modeling that bridges classical linear dynamical systems and modern deep learning, offering both expressive power and theoretical guarantees on stability and trainability. In recent years, multiple LRNN-based architectures have been proposed, each introducing distinct parameterizations, discretization schemes, and implementation constraints. However, existing implementations are fragmented across different software frameworks, often rely on framework-specific optimizations, and in some cases require custom CUDA kernels or lack publicly available code altogether. As a result, using, comparing, or extending LRNNs requires substantial implementation effort. To address this, we introduce $\texttt{lrnnx}$, a unified software library that implements several modern LRNN architectures under a common interface. The library exposes multiple levels of control, allowing users to work directly with core components or higher-level model abstractions. $\texttt{lrnnx}$ aims to improve accessibility, reproducibility, and extensibility of LRNN research and applications. We make our code available under a permissive MIT license.

Abstract (translated)

线性递归神经网络(LRNN)提供了一种结构化的方法来建模序列,它连接了经典的线性动态系统和现代深度学习技术,既具有表达能力又能保证稳定性和可训练性的理论保障。近年来,提出了多种基于LRNN的架构,每个架构都引入了不同的参数设置、离散化方案以及实现约束条件。然而,现有的实现方式分散在不同的软件框架中,常常依赖于特定框架的优化,并且有些需要定制CUDA内核或根本没有公开可用的代码。因此,使用、比较或扩展LRNN都需要大量的实现工作。 为解决这一问题,我们引入了$\texttt{lrnnx}$,这是一个统一的软件库,实现了几种现代的LRNN架构,在一个通用接口下进行操作。该库暴露了多个级别的控制,使用户可以直接与核心组件或更高层次的模型抽象进行交互。$\texttt{lrnnx}$旨在提高LRNN研究和应用的可访问性、重现性和扩展性。我们以宽松的MIT许可证提供代码。 --- 这段文本主要介绍了线性递归神经网络(LRNN)及其在现代深度学习中的重要地位,同时指出了当前实现这些模型时面临的一些挑战,并提出了一个名为$\texttt{lrnnx}$的新软件库来解决这些问题,通过统一接口和多层级控制提高了研究的可访问性和重现性。

URL

https://arxiv.org/abs/2602.08810

PDF

https://arxiv.org/pdf/2602.08810.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot