pub fn kept_pools(valid: &[u8], dims: DsaDims, seq: usize) -> Vec<i32>
2026-09-25: The original ids of the pools pool_states keeps. Computed from padding and sequence length alone.
pool_states