Shufan Li*

Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data

"In recent years, Transformers have become the de-facto architecture for sequence modeling on text and multi-dimensional data, such as images and video. However, the use of self-attention layers in a Transformer incurs prohibitive compute and memory complexity that scales quadratically w.r.t. the se…