google-deepmind / penzai

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

Alternatives and similar repositories for penzai:

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