Grokking as the transition from lazy to rich training dynamics - Kumar, Tanishq; Bordelon, Blake; Gershman, Samuel; Pehlevan, Cengiz (International Conference on Learning Representations, 2024)