Skip to content

Treescope 0.1.4

Compare
Choose a tag to compare
@danieldjohnson danieldjohnson released this 06 Sep 02:18
· 5 commits to main since this release

Bug fixes and improvements:

  • Figure rendering and automatic visualization:
    • Improved iframe size detection logic
    • Added ability to collapse autovisualizer IPythonVisualization outputs when replace=False
    • Reduced iframe duplication when using Treescope figures inside autovisualizers
    • Ensured DOM nodes are preserved when substituting deferred renderings to avoid breaking event handlers in figures
  • Added support for rendering JAX arrays with PmapSharding shardings
  • Improved support for rendering PyTorch tensors:
    • Turned off summarization for tensors on the "meta" device, since they have no data
    • Added support for float16 and bfloat16 tensors
    • Added device annotations in tensor summaries
  • Added default support for collections.UserDict and subclasses