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