Grokking as the Transition from Lazy to Rich Training Dynamics