openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

Apache License 2.0

2.64k stars 418 forks source link

#sdy add JAX Shardy support for memories. #17891

Closed copybara-service[bot] closed 3 days ago

copybara-service[bot] commented 1 week ago

sdy add JAX Shardy support for memories.