Penzai 0.2.3

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.