Display

Display objects using Treescope pretty-printer.

Abstract Signature:

Display(args)

PyTorch

API: print
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: print
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.nnx.display
Strategy: Direct Mapping
Official Docs ↗