PR #4239 Update Flax NNX Scale Up SPMD guide

Update Flax NNX Scale Up SPMD guide #4239

copybara-service merged 1 commit into google:main from 8bitmp3:upgrade-nnx-scale-up

IvyZX approved these changes on 2024-10-01

8bitmp3 force pushed 1 year ago

Update Flax NNX Scale Up SPMD guide

2bf7f9d1

8bitmp3 force pushed to 2bf7f9d1 1 year ago

8bitmp3 assigned

IvyZX 1 year ago

8bitmp3 assigned

8bitmp3 1 year ago

8bitmp3 added pull ready

copybara-service merged 7c5afdbf into main 1 year ago

8bitmp3 deleted the upgrade-nnx-scale-up branch 1 year ago

Reviewers

IvyZX

Assignees

IvyZX

8bitmp3

Labels

pull ready

Milestone

No milestone