| {"item_handlers": {"items": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "iter": "grain._src.python.checkpoint_handlers.CheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1771641984274074416, "commit_timestamp_nsecs": 1771641990503217302, "custom_metadata": {}} |