Comment by skadamat

If folks want a lower dimensional version of this for their own models, I'm a big fan of the Netron library for model architecture visualization.

Wrote about it here: https://about.xethub.com/blog/visualizing-ml-models-github-n...