Implementation of paper "Do Wide and Deep Networks Learn the Same Things? Uncovering How Neural Network Representations Vary with Width and Depth" by Thao Nguyen and Maithra Raghu and Simon Kornblith. Please find my post explaining the implementation on Medium/TDS.
The current implementation focuses on creating similar figures as in the paper: