xuhaoran

Support variable-length sequences for mamba block with position indices

2

Enable the mamba block to support variable-length sequence inputs using positional encoding. Passing Positional Indices results in negligible performance loss for the mamba block. For common variable-length sequence distributions, performance...