Add state_batch_indices to mamba3_step_fn (paged-inference state pools, parity with selective_state_update) - #971
Open
jgcb00 wants to merge 2 commits into
Open
Add state_batch_indices to mamba3_step_fn (paged-inference state pools, parity with selective_state_update)#971jgcb00 wants to merge 2 commits into
state_batch_indices to mamba3_step_fn (paged-inference state pools, parity with selective_state_update)#971jgcb00 wants to merge 2 commits into