"Treescope was originally developed as the pretty-printer for the Penzai"<p>Penzai: JAX research toolkit for building, editing, and visualizing neural nets
<a href="https://news.ycombinator.com/item?id=40107007">https://news.ycombinator.com/item?id=40107007</a>