PyTree prettyprinting #2
Description
Current manual implementation of strfmt for PyTrees in tree_util is messy at best. Switch to https://docs.kidger.site/equinox/api/pretty-printing/
{{ message }}
Current manual implementation of strfmt for PyTrees in tree_util is messy at best. Switch to https://docs.kidger.site/equinox/api/pretty-printing/