xla
Revive "Modularize zero step function and make it customizable" #7233
#8332
Merged
Merged
Revive "Modularize zero step function and make it customizable" #7233 #8332
jeffhataws
merged
5 commits into
master
from
jeffhataws_revive_zero1_modularize_pr
Modularize zero step function and make it customizable (#7233)
883fafd9
small fix to pass sharding_scheme instead of **kwargs
6abb432f
make method signatures consistent with oss
fe387956
Lint fixes
a69351bb
Fixing missing sharding_scheme param (#8341)
946fa8ab
hgt312
approved these changes
on 2024-11-01
mario-aws
approved these changes
on 2024-11-01
Login to write a write a comment.
Login via GitHub