penzai
A JAX research toolkit for building, editing, and visualizing neural networks.
Build, edit, and visualize neural networks using a toolkit that simplifies working with trained models.
Claim its page: indexed whatever its rank, translated into six languages, and enriched with what you write yourself.
Get an email alert on its next release or when it starts trending — never miss the moment.
Free · no card · unsubscribe anytimeA JAX research toolkit for building, editing, and visualizing neural networks.
penzai has 1.9k stars on GitHub. It has been forked 71 times. penzai is written mainly in Python. It has been in active development since 2024. penzai is available under the Apache-2.0 license. Its main topics are fine-tuning, interpretability, jax, neural-networks.
What is penzai?
A JAX research toolkit for building, editing, and visualizing neural networks.
Is penzai open source?
penzai is an open-source project. It is released under the Apache-2.0 license.
Is penzai free?
Yes. penzai is free and open source — you can use, modify and self-host it.
What license does penzai use?
penzai is available under the Apache-2.0 license.
What language is penzai written in?
penzai is written mainly in Python.
Add this live badge to your README — your GitHub stars and directory rank, refreshed daily.
[](https://olud.ai/project/google-deepmind-penzai.html)
Measured from GitHub topics shared by both projects, weighted by how rare each topic is.