A JAX research toolkit for building, editing, and visualizing neural networks.
What is the google-deepmind/penzai GitHub project? Description: "A JAX research toolkit for building, editing, and visualizing neural networks.". Written in Python. Explain what it does, its main use cases, key features, and who would benefit from using it.
Question is copied to clipboard — paste it after the AI opens.
Clone via HTTPS
Clone via SSH
Download ZIP
Download master.zipReport bugs or request features on the penzai issue tracker:
Open GitHub Issues