orbax
orbax copied to clipboard
google
Use `ocp.future.CommitFutureAwaitingContractedSignals` in custom async checkpoint handlers and type handlers to support async directory creation.
Use ocp.future.CommitFutureAwaitingContractedSignals in custom async checkpoint handlers and type handlers to support async directory creation.
![copybara-service[bot] avatar](https://avatars.githubusercontent.com/in/44061?v=4 "Published by a pub.dev verified publisher"){kind=small}