Question: At which step, the loss spikes when without QK-norm?
Answer: 150k