google-deepmind / penzaiLinks

A JAX research toolkit for building, editing, and visualizing neural networks.
1,811Updated 2 months ago

Alternatives and similar repositories for penzai

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

Sorting: