PyPI · #2917 most downloaded on PyPI

treescope

0.1.101.9M downloads/mogoogle-deepmind/treescope ↗New releases in your daily digest, changelog included.

Treescope: An interactive HTML pretty-printer for ML research in IPython notebooks.

Last release 1 years ago
08 Aug 2025
Release timing varies
gaps range from 9 days to 6 months
Nearly every release is documented
notes for 11 of 11 stable releases
Nothing withdrawn
no release was ever pulled
2 years old
11 releases · first in 2024

Release timeline

11 releases since 2024

202420252026

2024

Releases

0.1.108 Aug 2025
Fixes incompatibilities with JAX v0.7.0
Fixes incompatibilities with JAX v0.7.0
Open source →
0.1.917 Feb 2025
New configuration option abbreviation_threshold that abbreviates collapsed objects to prevent extremely long lines (idea from #53, implemented in #56)
New features

New configuration option abbreviation_threshold that abbreviates collapsed objects to prevent extremely long lines (idea from #53, implemented in #56)

You can use it by passing abbreviation_threshold=1 to treescope.basic_interactive_setup or by modifying the config setting manually (see the docs or the PR for details).

Torch tensors and parameters now include requires_grad info (from #51, implemented in #59)

Treescope now supports rendering Pydantic BaseModel and Hydra/OmegaConf DictConfig/ListConfig out of the box (implemented in #60)
Open source →

Fixed an error when rendering new-style JAX PRNG keys (with extended dtypes) (#52, #54)
Improved how Treescope renderings appear in the VSCode embedded Jupyter notebook editor and in dark mode viewers:
- Added workaround for false positives in VSCode's detection of empty output cells, which was accidentally hiding some outputs
- Changed white background color to fill the line width, improving appearance when running Treescope inside a dark-mode theme
Fixed documentation pages not rendering Treescope outputs on the landing page, due to a too-old version of Penzai being imported

Open source →

0.1.716 Dec 2024

Added handlers for the new C-based JAX pytree classes (https://github.com/jax-ml/jax/pull/24906), which are no longer dataclasses. The new handlers pr

New features

Added handlers for the new C-based JAX pytree classes (https://github.com/jax-ml/jax/pull/24906), which are no longer dataclasses. The new handlers preserve the original dataclass-like rendering behavior and support upgraded JAX versions.

Other changes

Updated uv lockfile to use newer package versions when setting up the development environment and running tests.
Silenced some warnings that were triggering in tests.

Open source →

0.1.621 Nov 2024

Rendering JAX arrays is now much faster, especially for small device arrays

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.

Open source →

0.1.513 Sept 2024Deprecated

Removed check for deprecated named_shape attribute on jax.ShapeDtypeStruct, since this attribute is being removed from JAX.

Bug fixes and improvements:

Removed check for deprecated named_shape attribute on jax.ShapeDtypeStruct, since this attribute is being removed from JAX.
Added a "part" attribute to the root rendering element to allow customizing CSS styles via the CSS shadow parts system (#38).

Open source →

0.1.46 Sept 2024

Figure rendering and automatic visualization:

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

Open source →

0.1.327 Aug 2024

render_array and ArrayAutovisualizer now accept a pixels_per_cell argument to customize the initial zoom level of array renderings.

New features

render_array and ArrayAutovisualizer now accept a pixels_per_cell argument to customize the initial zoom level of array renderings.
Array renderings now adapt their size and rendering options based on device resolution, to ensure renderings are crisp when the window is zoomed in or when used on a high-DPI display.

Bug fixes and improvements:

Fixed a type error with prefers_row/prefers_column in ArrayAutovisualizer.
Simplified torch array adapter to not require named axis support or isposinf/isneginf primitives, allowing it to be used with the "mps" backend (Apple silicon).
Treescope now correctly catches and handles errors that occur during deferred rendering or inside custom __repr__ implementations.

Open source →

0.1.212 Aug 2024

Fixed rendering of PyTorch GPU tensors

Bug fixes and improvements:

Fixed rendering of PyTorch GPU tensors (#19)
Treescope now avoids overriding the custom IPython format function _repr_mimebundle_ for types that define it

Documentation changes:

Fixed typos in README and docs home page (#22)
Added information about how to cite the accompanying whitepaper

Open source →

0.1.17 Aug 2024

Added the autovisualize keyword argument to treescope.show and treescope.display to allow customizing the automatic visualizer on a per-call basis, al

New features:

Added the autovisualize keyword argument to treescope.show and treescope.display to allow customizing the automatic visualizer on a per-call basis, along with a few more configuration options.
Added alternate forms %%autovisualize True and %%autovisualize False for the %%autovisualize magic.

Bug fixes and improvements:

Adjusted output wrapper to ensure that "Loading" indicators appear when the browser is loading a rendered output.
Fix a visual bug with torch.nn.Sequential rendering.

Documentation changes:

Added documentation of neural network support.
Fixed broken links.

Open source →

0.1.025 Jul 2024

Initial release of the standalone Treescope pretty-printer.

Treescope was originally designed as the pretty-printer for the Penzai neural network library, but has been moved to its own repository so that it can be installed and used independently. See the commit history for Penzai for earlier versions of Treescope.

Changes relative to the penzai.treescope submodule of Penzai release 0.1.5:

Moved all functions and methods out of the penzai.treescope namespace and into treescope.
Renamed extension protocol method from __penzai_repr__ to __treescope_repr__.
Added a global handler registry, to enable registering handlers for existing types.
Redesigned the array visualization system in terms of an NDArrayAdapter protocol, enabling it to support non-JAX arrays.
Added handlers for PyTorch tensors and PyTorch neural network modules.
Moved implementation details of the internal intermediate representation into an _internal subpackage, and added new public construction methods for various rendering components.
Removed some complex features that were rarely used (copy-paste fallbacks and between-node hyperlinks).

Open source →