google-deepmind/penzai
A JAX research toolkit for building, editing, and visualizing neural networks.
License & maintenance
Apache-2.0 is a permissive license: free to use commercially with attribution.
Last code push 16 months ago. Archived: read-only, no further updates.