PyPI Stats

Search

All packages
Top packages

Track packages

penzai


PyPI page
Home page
Author: None
Summary: Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
Latest version: 0.1.2
Required dependencies: absl-py | equinox | jax | numpy | ordered_set | typing_extensions
Optional dependencies: flax | ipython | jax | jupyter | matplotlib | myst-nb | optax | orbax-checkpoint | palettable | pyink | pylint | sphinx | sphinx-book-theme | sphinx-collections | sphinx-hoverxref | sphinx_contributors | sphinxcontrib-katex

Downloads last day: 42
Downloads last week: 318
Downloads last month: 3,329