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 139 days ago
tengyifei tengyifei force pushed from cf50ec59 to acedcbec 139 days ago
tengyifei Improve `assume_pure` SPMD functionality
7b045f79
tengyifei tengyifei force pushed from acedcbec to 7b045f79 139 days ago
tengyifei tengyifei requested a review from qihqi qihqi 139 days ago
tengyifei tengyifei marked this pull request as ready for review 139 days ago
tengyifei Avoid expensive EinsumLinear initialization
c4591a52
qihqi
qihqi approved these changes on 2025-06-16
tengyifei tengyifei merged 0be0fbdf into master 138 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone