scores = q @ k.T + relative_position_bias