google-deepmind/penzai issues and pull requests

#106 - Update default package versions used by uv.

Pull Request - State: closed - Opened by danieldjohnson about 2 months ago

#104 - Check for modified config attributes during HuggingFace conversion

Pull Request - State: closed - Opened by danieldjohnson about 2 months ago

#103 - Alternative data model for NamedArray using tuple[str | None, ...] for axis names

Issue - State: closed - Opened by shoyer about 2 months ago

#102 - Fix RoPE theta in config

Pull Request - State: closed - Opened by Fedzbar 2 months ago - 1 comment

#101 - Fix RoPE wavelength

Pull Request - State: closed - Opened by Fedzbar 2 months ago - 1 comment

#100 - Disable the treescope renderer test temporarily as it breaks if JAX key classes change.

Pull Request - State: closed - Opened by copybara-service[bot] 2 months ago - 3 comments

#99 - Experiment with using out_axes=-1 inside nmap's vmap

Pull Request - State: open - Opened by shoyer 2 months ago

#98 - Fix documentation build: pin packaging==24.1, install with uv.

Pull Request - State: closed - Opened by danieldjohnson 3 months ago

#97 - Fix VSCode autoformat config to avoid bad Python version for pyink.

Pull Request - State: closed - Opened by danieldjohnson 3 months ago

#96 - Switch LayerStackGetAttrKey to a custom dataclass type.

Pull Request - State: closed - Opened by danieldjohnson 3 months ago

#95 - Add autoformatting check for test files.

Pull Request - State: closed - Opened by danieldjohnson 3 months ago - 1 comment

#94 - Issue on page /index.html

Issue - State: closed - Opened by theicehouse 3 months ago - 2 comments

#93 - Make Penzai `LayerStackGetAttrKey` a pure Python class instead of a dataclass.

Pull Request - State: closed - Opened by copybara-service[bot] 3 months ago - 1 comment

#92 - More informative structure mismatch error message in Linear layer

Pull Request - State: closed - Opened by amifalk 3 months ago - 3 comments

#91 - [Do Not Merge] PR sync test

Pull Request - State: closed - Opened by danieldjohnson 3 months ago

#90 - Configure uv, pyink, and pytype, and fix lint/type errors.

Pull Request - State: closed - Opened by danieldjohnson 3 months ago

#89 - FR: add error prefix to check_structure call in pz.nn.Linear

Issue - State: closed - Opened by amifalk 4 months ago

#87 - Release v0.2.2

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#86 - Removed `named_shape=` from `jax.core.ShapedArray` and `jax.ShapeDtypeStruct`

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#85 - Add protocol methods to Penzai variables for library interop.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#84 - Catch exceptions during deferred selection rendering.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#83 - Adjust unittest action config to avoid duplicate triggers.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#82 - Determinism in Gemma model

Issue - State: closed - Opened by daralthus 6 months ago - 1 comment

#81 - Configure unit tests to run on pull requests.

Pull Request - State: closed - Opened by copybara-service[bot] 6 months ago

#80 - FR: expose device/sharding argument for NamedArray.wrap

Issue - State: open - Opened by amifalk 6 months ago - 2 comments

#79 - Add citation info for Penzai/Treescope paper.

Pull Request - State: closed - Opened by copybara-service[bot] 6 months ago

#78 - Release v0.2.1

Pull Request - State: closed - Opened by copybara-service[bot] 6 months ago

#77 - Add support for Gemma 2 models.

Pull Request - State: closed - Opened by copybara-service[bot] 6 months ago