Reminds me of Grokker https://www.researchgate.net/figure/Grokker-Visualization-of...