google-deepmind / penzaiView on GitHub
A JAX research toolkit for building, editing, and visualizing neural networks.
1,893Jun 22, 2025Updated last year

Alternatives and similar repositories for penzai

Users that are interested in penzai are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?