Package search
penzai
- Summary: Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
- Author:
- 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-18T15:06:47
- Latest release: 0.2.5 on 2025-04-08T05:43:58
| 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 |