PR #4220 Flax NNX GSPMD guide

Flax NNX GSPMD guide #4220

copybara-service merged 1 commit into google:main from IvyZX:jit-guide

IvyZX requested a review from

cgarciae 1 year ago

cgarciae commented on 2024-09-24

cgarciae approved these changes on 2024-09-24

IvyZX force pushed 1 year ago

Flax gspmd guide

9b41f61c

IvyZX force pushed to 9b41f61c 1 year ago

IvyZX added pull ready

copybara-service merged 9c86396c into main 1 year ago

Reviewers

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone