marcusrogerio / ai.penzai Goto Github PK
View Code? Open in Web Editor NEWThis project forked from google-deepmind/penzai
A JAX research toolkit for building, editing, and visualizing neural networks.
Home Page: https://penzai.readthedocs.io/
License: Apache License 2.0