xla
Improve `assume_pure` SPMD functionality
#9360
Merged

Improve `assume_pure` SPMD functionality #9360

tengyifei merged 2 commits into master from yifeit/nested-pspec
tengyifei
tengyifei tengyifei force pushed from 2604178e to cf50ec59 1 year ago
tengyifei tengyifei force pushed from cf50ec59 to acedcbec 1 year ago
tengyifei Improve `assume_pure` SPMD functionality
7b045f79
tengyifei tengyifei force pushed from acedcbec to 7b045f79 1 year ago
tengyifei tengyifei requested a review from qihqi qihqi 1 year ago
tengyifei tengyifei marked this pull request as ready for review 1 year ago
tengyifei Avoid expensive EinsumLinear initialization
c4591a52
qihqi
qihqi approved these changes on 2025-06-16
tengyifei tengyifei merged 0be0fbdf into master 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone