This is a delightful exposition of grokking. I'm very glad to see such a beautiful phenomenom deserve a similarly striking visualisation.

What are you writing these visualisations in?