Download Latest Version Penzai 0.2.5 source code.tar.gz (905.2 kB)
Email in envelope

Get an email when there's a new version of Penzai

Home / v0.2.3
Name Modified Size InfoDownloads / Week
Parent folder
Penzai 0.2.3 source code.tar.gz 2024-11-21 905.1 kB
0
Penzai 0.2.3 source code.zip 2024-11-21 1.1 MB
0
README.md 2024-11-21 811 Bytes
0
Totals: 3 Items   2.0 MB 0

New features and improvements

  • Named array shape mismatch errors inside Linear layers now give a more informative error message based on the parameter name, if possible (#92).
  • The custom PyTree key LayerStackGetAttrKey used by penzai.nn.layer_stack.LayerStack no longer directly subclasses jax.tree_util.GetAttrKey (#96), making it robust to future planned changes in JAX.

Other changes

  • Project now includes a uv lockfile and can be easily installed using the uv package manager.
  • Checks for formatting (via pyink), linting (via pylint), and type annotations (via pytype) are now run on every PR.
  • pyink and pylint are configured to automatically run when editing the project in VSCode.
  • Made minor tweaks to documentation configuration to avoid a build error.
Source: README.md, updated 2024-11-21