PyPI page
Home page
Author:
None
Summary:
Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
Latest version:
0.2.0
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
|
palettable
|
pyink
|
pylint
|
pytest
|
sphinx
|
sphinx-book-theme
|
sphinx-collections
|
sphinx-hoverxref
|
sphinx_contributors
|
sphinxcontrib-katex
|
torch
|
transformers
Downloads last day:
175
Downloads last week:
1,348
Downloads last month:
5,887