### 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)

- 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

### 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.

### 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.

## 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).

### 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

### 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.

### 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