compute QK^T
store matrix
apply softmax