Skip to content

contrib: add sharding tests for contrib optimizers#1714

Open
irhyl wants to merge 3 commits into
google-deepmind:mainfrom
irhyl:contrib/sharding-tests
Open

contrib: add sharding tests for contrib optimizers#1714
irhyl wants to merge 3 commits into
google-deepmind:mainfrom
irhyl:contrib/sharding-tests

contrib: drop jit sharding checks, eager init/update is sufficient

5889eaa
Select commit
Loading
Failed to load commit list.