Skip to content

Treescope 0.1.6

Compare
Choose a tag to compare
@danieldjohnson danieldjohnson released this 21 Nov 23:06
· 12 commits to main since this release

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.