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