search

google / orbax

Orbax provides common checkpointing and persistence utilities for JAX users
https://orbax.readthedocs.io/
Apache License 2.0
305 stars 36 forks source link

Update CompositeCheckpointHandler.metadata() to return StepMetadata. #1398

Open copybara-service[bot] opened 17 hours ago

copybara-service[bot] commented 17 hours ago

Update CompositeCheckpointHandler.metadata() to return StepMetadata.