flax
41c9304c
- Make Flax checkpointing use `sharding` instead of extracting mesh and pspecs. Also improved some warning messages.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
3 years ago
Make Flax checkpointing use `sharding` instead of extracting mesh and pspecs. Also improved some warning messages. PiperOrigin-RevId: 518675385
Author
IvyZX
Committer
a-googler
Parents
c2da6f90
Loading