orbax
orbax copied to clipboard
google
Refactor Flax checkpointing to closer integrate with Orbax class structure. We can duplicate Flax's behavior using Orbax.
Refactor Flax checkpointing to closer integrate with Orbax class structure. We can duplicate Flax's behavior using Orbax.
![copybara-service[bot] avatar](https://avatars.githubusercontent.com/in/44061?v=4 "Published by a pub.dev verified publisher"){kind=small}