Last released Sep 13, 2024
Treescope: An interactive HTML pretty-printer for ML research in IPython notebooks.
Last released Sep 11, 2024
Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
Supported by