Remove `dataclasses` from dependency

PiperOrigin-RevId: 488800131

Increase Orbax version.

PiperOrigin-RevId: 487899957

Increase orbax.checkpoint version.

PiperOrigin-RevId: 487329098

Increase Orbax version.

PiperOrigin-RevId: 485621170

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

PiperOrigin-RevId: 482004157

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

…used to represent both GDA and jax.Array. Restoration logic automatically figures out whether to return jax.Array or GDA depending on the setting of jax.config.

PiperOrigin-RevId: 477502612

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

…file names in structure rather than absolute paths.

PiperOrigin-RevId: 475845023

Make cleanup_tmp_directories method a class method for CheckpointMana…

…ger to allow for customization.

PiperOrigin-RevId: 474902053

Increase Orbax version.

PiperOrigin-RevId: 473357218

Add check_for_errors method in CheckpointManager to check for async e…

…rrors.

PiperOrigin-RevId: 471023594