Last released May 3, 2024
Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
Supported by