flax
fde3b2e8 - Remove spmd_lowering from MeshComputation and do more cleanups after pmap, PmapSharding is gone
Commit
View On GitHub
Commit
9 days ago
Remove spmd_lowering from MeshComputation and do more cleanups after pmap, PmapSharding is gone PiperOrigin-RevId: 886581793
Author
hawkinsp hawkinsp
Committer
a-googler a-googler
Parents
ee6d3c58
FAQ Terms Privacy Refunds Impressum
Loading