Allow registering an item in `DefaultCheckpointHandlerRegistry` without providing the actual handler, as long as the provided args correspond to a globally registered handler. This allows for slightly reduced verbosity if we just want to ensure an association between an item name and args/handler.

google / orbax

Orbax provides common checkpointing and persistence utilities for JAX users

https://orbax.readthedocs.io/

Apache License 2.0

307 stars 36 forks source link

Allow registering an item in `DefaultCheckpointHandlerRegistry` without providing the actual handler, as long as the provided args correspond to a globally registered handler. This allows for slightly reduced verbosity if we just want to ensure an association between an item name and args/handler. #1208

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 2 months ago

Allow registering an item in DefaultCheckpointHandlerRegistry without providing the actual handler, as long as the provided args correspond to a globally registered handler. This allows for slightly reduced verbosity if we just want to ensure an association between an item name and args/handler.