Add state_batch_indices to mamba3_step_fn (paged-inference state pools, parity with selective_state_update)#971

jgcb00 wants to merge 1 commit into

state-spaces:mainstate-spaces/mamba:mainfrom

jgcb00:mamba3-state-poolsjgcb00/mamba:mamba3-state-poolsCopy head branch name to clipboard