Package profile
penzai
- Summary: Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
- Author: The Penzai Authors <penzai-dev@google.com>
- Homepage: https://github.com/google-deepmind/penzai
- Source: https://github.com/google-deepmind/penzai (Repo profile)
- Number of releases: 12
- First release: 0.1.0 on 2024-04-18
- Latest release: 0.2.5 on 2025-04-08
Releases
PyPI Downloads
Dependencies
Penzai has 29 dependencies, 23 of which optional.| Dependency | Optional |
|---|---|
| absl-py | false |
| jax | false |
| numpy | false |
| ordered_set | false |
| treescope | false |
| typing_extensions | false |
| flax | true |
| ipython | true |
| jupyter | true |
| matplotlib | true |
| myst-nb | true |
| myst-parser | true |
| optax | true |
| orbax-checkpoint | true |
| packaging | true |
| palettable | true |
| pyink | true |
| pylint | true |
| pytest | true |
| pytype | true |
| setuptools | true |
| sphinx | true |
| sphinx_contributors | true |
| sphinx-book-theme | true |
| sphinx-collections | true |
| sphinx-hoverxref | true |
| sphinxcontrib-katex | true |
| torch | true |
| transformers | true |