penzai
by google-deepmind · GitHub
A JAX research toolkit for building, editing, and visualizing neural networks.
⭐ Stars
1.9k
🍴 Forks
71
📜 License
Apache-2.0
Commercial use OK📅 Created
2024
🔄 Last commit
1 yr ago
🏷️ Category
fine-tuning
💻 Language
You maintain this project?
Claim its page: indexed whatever its rank, translated into six languages, and enriched with what you write yourself.
📈 Star history
2026-07-202026-07-21
📄 About
A JAX research toolkit for building, editing, and visualizing neural networks.
❓ Frequently asked questions
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.
🏅 Maintainer of this project?
Add this live badge to your README — your GitHub stars and directory rank, refreshed daily.
[](https://opensourceai.tech/project/google-deepmind-penzai.html)
🧬 Shares DNA with🧬 View the DNA map →
axlearn
An Extensible Deep Learning Library
★ 2.4k · deep-learning
sharesjax
LlamaFactory
Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)
★ 73.4k · agent
unsloth
Unsloth is a local UI for training and running Gemma 4, Qwen3.6, DeepSeek, Kimi, GLM and other…
★ 68.6k · agent
Measured from GitHub topics shared by both projects, weighted by how rare each topic is.