← Search

Aakash Lahoti

3 accepted papers

2026

Mamba-3: Improved Sequence Modeling using State Space Principles

ICLR 2026oral

The recent scaling of test-time compute for LLMs has restricted the practical deployment of models to those with strong capabilities that can generate high-quality outputs in an inference-efficient manner. While current Transformer-based models are the standard, their quadratic compute and linear me…

Cited by 0SourcecodeScholar
2024

Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers

NeurIPS 2024poster

A wide array of sequence models are built on a framework modeled after Transformers, comprising alternating sequence mixer and channel mixer layers. This paper studies a unifying *matrix mixer* view of sequence mixers that can be conceptualized as a linear map on the input sequence. This framework…

2024

Role of Locality and Weight Sharing in Image-Based Tasks: A Sample Complexity Separation between CNNs, LCNs, and FCNs

ICLR 2024spotlight

Vision tasks are characterized by the properties of locality and translation invariance. The superior performance of convolutional neural networks (CNNs) on these tasks is widely attributed to the inductive bias of locality and weight sharing baked into their architecture. Existing attempts…

Cited by 1SourcePDFScholar