google-deepmind / penzaiView on GitHub
A JAX research toolkit for building, editing, and visualizing neural networks.
1,869Jun 22, 2025Updated 8 months ago

Alternatives and similar repositories for penzai

Users that are interested in penzai are comparing it to the libraries listed below

Sorting:

Are these results useful?