xla
Add shard_map wrapper
#7670
Merged

Loading