Jiajun Zhu

Baleen: Self‑Interpretable, Robust SSMs with Stochastic Selective Memory

Rethinking Addressing in Language Models via Contextualized Equivariant Positional Encoding

Rethinking Addressing in Language Models via Contextualized Equivariant Positional Encoding

Understanding and Mitigating Bottlenecks of State Space Models through the Lens of Recency and Over-smoothing