google-deepmind/treescope v0.1.6

Treescope 0.1.6

Repository: google-deepmind/treescope

Tag: v0.1.6

Published: 2024-11-21T23:06:25Z

Prerelease: no

Release notes:

New features

• Rendering JAX arrays is now much faster, especially for small device arrays (#40)

Bug fixes and improvements

• Fixed incorrect bloat16 array formatting (#43)
• Fixed a number of typos (#44)
• Fixed render_array to correctly support re-ordering rows and columns to match documented behavior (#49)
• Test no longer depend on the exact representation of jax.tree_util.GetAttrKey, making it robust to future planned changes in JAX (#48)

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.