Penzai: JAX research toolkit for building, editing, and visualizing neural nets https://news.ycombinator.com/item?id=40107007

> Treescope is an interactive HTML pretty-printer and N-dimensional array ("tensor") visualizer designed for machine learning and neural networks research in IPython notebooks

Its major feature, from the readme, seems to be graphical visualisation. There's a screenshot a page or two down. It's a little more than an "Interactive HTML Pretty-Printer" which I interpreted as a HTML formatter / syntax highlighter you could interact with.