jax

29261d2d - `pretty_printer.Doc` now uses `use_cpp_class`

`pretty_printer.Doc` now uses `use_cpp_class` This pattern is used by many existing JAX internals as an alternative to monkey-patching individual methods. PiperOrigin-RevId: 895277310