google-deepmind / penzaiLinks

A JAX research toolkit for building, editing, and visualizing neural networks.
1,780Updated last month

Alternatives and similar repositories for penzai

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

Sorting: