/penzai

A JAX research toolkit for building, editing, and visualizing neural networks.

Primary LanguagePythonApache License 2.0Apache-2.0

Watchers