Add compression metadata and telemetry helpers to Orbax checkpoint. - #3561

Warning: Code scanning cannot determine the alerts introduced by this pull request, because 1 configuration present on refs/heads/main was not found:

Default setup

• ❓  /language:python

View all branch alerts.