flax
Guide for Flax partitioning (auto SPMD) API
#2730
Merged

Guide for Flax partitioning (auto SPMD) API #2730

copybara-service merged 3 commits into google:main from IvyZX:pjit
IvyZX
IvyZX IvyZX requested a review from levskaya levskaya 3 years ago
IvyZX IvyZX assigned IvyZX IvyZX 3 years ago
review-notebook-app
IvyZX IvyZX force pushed 3 years ago
marcvanzee marcvanzee requested a review from marcvanzee marcvanzee 3 years ago
marcvanzee marcvanzee requested a review from 8bitmp3 8bitmp3 3 years ago
marcvanzee
levskaya
levskaya commented on 2022-12-17
IvyZX IvyZX force pushed 3 years ago
IvyZX IvyZX force pushed 3 years ago
IvyZX IvyZX force pushed 3 years ago
IvyZX
marcvanzee
marcvanzee requested changes on 2022-12-19
IvyZX IvyZX force pushed 3 years ago
IvyZX IvyZX force pushed 3 years ago
IvyZX IvyZX requested a review from marcvanzee marcvanzee 3 years ago
IvyZX
8bitmp3
8bitmp3 commented on 2022-12-20
marcvanzee
marcvanzee approved these changes on 2022-12-20
andsteing andsteing requested a review from andsteing andsteing 3 years ago
andsteing
andsteing commented on 2022-12-20
cgarciae
marcvanzee
cgarciae
8bitmp3
IvyZX create pjit spmd guide doc
023511b6
IvyZX IvyZX force pushed to 023511b6 3 years ago
IvyZX
andsteing
andsteing commented on 2022-12-22
andsteing
andsteing commented on 2022-12-22
andsteing
andsteing approved these changes on 2022-12-22
8bitmp3 Update Flax pjit guide notebook
dcf5517e
8bitmp3
8bitmp3
8bitmp3 approved these changes on 2022-12-22
IvyZX pjit guide final touches
4920bd85
IvyZX IvyZX added pull ready
copybara-service copybara-service merged 0453d38f into main 3 years ago

Login to write a write a comment.

Login via GitHub

Assignees
Labels
Milestone