Grokking as a First Order Phase Transition in Two Layer Networks - Rubin, Noa; Seroussi, Inbar; Ringel, Zohar (International Conference on Learning Representations, 2024)