/penzai

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

Primary LanguagePythonApache License 2.0Apache-2.0

Watchers

No one’s watching this repository yet.