Skip to content

Penzai 0.1.1
Compare
Choose a tag to compare
@danieldjohnson danieldjohnson released this 03 May 22:19
· 64 commits to main since this release
v0.1.1
4d08e2b

Penzai 0.1.1 fixes a number of bugs and adds a few convenience features.

New features:

  • NamedArray can now be converted automatically to a JAX array without unwrapping it if it has no more named axes (e.g. after calling untag).
  • Lifting JAX functions to operate over NamedArrays can be accomplished with less boilerplate using penzai.toolshed.auto_nmap: if you set njnp = auto_nmap.wrap_module(jnp), you can use njnp.{method} instead of pz.nx.nmap(jnp.{method}).

Bug fixes and improvements:

  • Bugfix: Array data in array visualization tooltips now correctly accounts for slider offsets.
  • Improvements to compatibility of penzai.treescope renderings:
    • Copy-path buttons no longer require internet access.
    • Pretty-printed outputs should now appear in the correct location in JupyterLab.
    • Perfomance improvements for non-Chrome browsers.

Documentation changes:

  • Fixed typos and formatting issues.
  • Fixed notebook output display bug in Firefox.
  • Source links in ReadTheDocs now link to a specific commit.
  • Cross-page links in ReadTheDocs now support hover tooltip previews.
Assets 2
Loading