PyPI page
Home page
Author:
None
Summary:
Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
Latest version:
0.2.4
Required dependencies:
absl-py
|
jax
|
numpy
|
ordered_set
|
treescope
|
typing_extensions
Optional dependencies:
flax
|
ipython
|
jax
|
jupyter
|
matplotlib
|
myst-nb
|
myst-parser
|
optax
|
orbax-checkpoint
|
packaging
|
palettable
|
pyink
|
pylint
|
pytest
|
pytype
|
setuptools
|
sphinx
|
sphinx-book-theme
|
sphinx-collections
|
sphinx-hoverxref
|
sphinx_contributors
|
sphinxcontrib-katex
|
torch
|
transformers
Downloads last day:
1
Downloads last week:
484
Downloads last month:
2,487