Comment by felixr

"Treescope was originally developed as the pretty-printer for the Penzai"

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