penzai

penzai

google-deepmind

A JAX research toolkit for building, editing, and visualizing neural networks.

1.9k Stars
70 Forks
1.9k Watchers
Python Language
apache-2.0 License
100 SrcLog Score
Cost to Build
$27.40M
Market Value
$107.78M

Growth over time

4 data points  ·  2025-07-14 → 2026-04-26
Stars Forks Watchers
💬

How do you feel about this project?

Ask AI about penzai

Question copied to clipboard

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.

How to clone penzai

Clone via HTTPS

git clone https://github.com/google-deepmind/penzai.git

Clone via SSH

[email protected]:google-deepmind/penzai.git

Download ZIP

Download master.zip

Found an issue?

Report bugs or request features on the penzai issue tracker:

Open GitHub Issues