pub fn index_scores(
q: &[f32],
weights: &[f32],
pools: &Pools,
dims: DsaDims,
q_rows: usize,
) -> Vec<f32>Expand description
2026-09-25: Per-(query, pool) index score: sum_h weights[h] * relu(scale * q_h . key).
q is [q_rows, index_heads, index_head_dim]; weights is [q_rows, index_heads] and must
already carry the index_heads^-0.5 factor, as the golden generator applies it.