PackMamba: Efficient Processing of Variable-Length Sequences in Mamba Training
Haoran Xu, Ziqian Liu, Rong Fu, Zhongling Su, Zerui Wang, Zheng Cai, Zhilin Pei, Xingcheng Zhang
May, 2025
Abstract
PackMamba improves the efficiency of training Mamba models on variable-length sequences. It packs sequences and modifies the convolution and selective-scan operators to preserve sequence boundaries, reducing padding overhead while maintaining parallel execution.
Publication
In Computer Vision – ECCV 2024 Workshops, published in 2025

Ph.D. Student · Research Intern