flax
515393d8 - Make `restore_args_from_target` API use sharding by default.

Commit

2 years ago

Make `restore_args_from_target` API use sharding by default. PiperOrigin-RevId: 520441427

Author

IvyZX

Committer

a-googler

Parents