scores = Q @ K.T