xla
Improve `assume_pure` SPMD functionality
#9360
Merged

Commits
  • Improve `assume_pure` SPMD functionality
    tengyifei committed 1 year ago
  • Avoid expensive EinsumLinear initialization
    tengyifei committed 1 year ago
Loading