v0.0.22

Added

• Added CHANGELOG.md for version updates (additions and changes), ingested by
  auto-publish functionality.

v0.0.19

What's Changed

• [Orbax] Log unmatched keys from the new tree by @copybara-service in #147
• Fix auto-publish tool, replace setup.py with pyproject.toml standard, disable duplicate builds. by @copybara-service in #154
• Add support for tracking checkpoint metrics with Orbax in T5X. by @copybara-service in #153

Full Changelog: v0.0.18...v0.0.19

v0.0.18

What's Changed

• Add auto-publish tool for Orbax. by @copybara-service in #142

Full Changelog: v0.0.16...v0.0.18

Release

What's Changed

• Add update API for CheckpointManager. At the moment, this will only support updating metrics. We will consider whether to support updating any part of the checkpoint in the future. by @copybara-service in #137

Full Changelog: v0.0.15...v0.0.16

v0.0.15

What's Changed

• Add missing import to JAX gda_serialization. by @copybara-service in #131
• Increase orbax.checkpoint version. by @copybara-service in #135

Full Changelog: v0.0.14...v0.0.15

v0.0.14 Release

What's Changed

• Simplify handling of msgpack-serialized files by only saving the parameter name rather than the full ts.Spec. by @copybara-service in #101
• Add keep_period option in CheckpointManager to allow users to keep all checkpoints falling on that interval, even if they would otherwise be deleted. by @copybara-service in #120
• TypeHandler refactor of PytreeCheckpointHandler. At a high level, this will allow us to provide greater flexibility for user-defined types as PyTree leaves. by @copybara-service in #107
• Switch gfile usage to epath.Path for a few core files by @copybara-service in #117
• Rollback TypeHandler. by @copybara-service in #122
• TypeHandler refactor of PytreeCheckpointHandler. At a high level, this will allow us to provide greater flexibility for user-defined types as PyTree leaves. by @copybara-service in #123
• Increase Orbax version. by @copybara-service in #130

Full Changelog: v0.0.13...v0.0.14

Release

Support older versions of Python. (External colab uses 3.7).

PiperOrigin-RevId: 482004157

Release

v0.0.12

Remove `use_gda` flag and replace with `use_jax_array` only. This is …

Release

v0.0.11

Fix structure caching for enable_flax=False case by storing relative …

Release

v0.0.10

Make cleanup_tmp_directories method a class method for CheckpointMana…