PR #2730 Guide for Flax partitioning (auto SPMD) API

Guide for Flax partitioning (auto SPMD) API #2730

copybara-service merged 3 commits into google:main from IvyZX:pjit

IvyZX requested a review from

levskaya 3 years ago

IvyZX assigned

IvyZX 3 years ago

IvyZX force pushed 3 years ago

marcvanzee requested a review from

marcvanzee 3 years ago

marcvanzee requested a review from

8bitmp3 3 years ago

levskaya commented on 2022-12-17

IvyZX force pushed 3 years ago

marcvanzee requested changes on 2022-12-19

IvyZX force pushed 3 years ago

IvyZX requested a review from

marcvanzee 3 years ago

8bitmp3 commented on 2022-12-20

marcvanzee approved these changes on 2022-12-20

andsteing requested a review from

andsteing 3 years ago

andsteing commented on 2022-12-20

create pjit spmd guide doc

023511b6

IvyZX force pushed to 023511b6 3 years ago

andsteing commented on 2022-12-22

andsteing approved these changes on 2022-12-22

Update Flax pjit guide notebook

dcf5517e

8bitmp3 approved these changes on 2022-12-22

pjit guide final touches

4920bd85

IvyZX added pull ready

copybara-service merged 0453d38f into main 3 years ago

Reviewers

8bitmp3

andsteing

marcvanzee

levskaya

Assignees

IvyZX

Labels

pull ready

Milestone

No milestone