flax
Remove spmd_lowering from MeshComputation and do more cleanups after pmap, PmapSharding is gone
#5360
Open
Overview
Commits
1
Changes
View On GitHub

Remove spmd_lowering from MeshComputation and do more cleanups after pmap, PmapSharding is gone #5360

copybara-service wants to merge 1 commit into main from test_886581793
copybara-service
copybara-service copybara-service force pushed from 72ebe1a7 to ad3f54e7 14 days ago
hawkinsp Remove spmd_lowering from MeshComputation and do more cleanups after …
fde3b2e8
copybara-service copybara-service force pushed from ad3f54e7 to fde3b2e8 14 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum