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…

Release

Increase Orbax version.

PiperOrigin-RevId: 473357218

Release

v0.0.8

Add check_for_errors method in CheckpointManager to check for async e…

Release

v0.0.7

Small change in `should_save` behavior which avoids always saving a c…

Release

Increase Orbax version.

PiperOrigin-RevId: 466145254

Release

Increase version to expose AsyncCheckpointer.

PiperOrigin-RevId: 463921320