xla

3496de89 - [SPMD] construct auto-sharding mesh ids based on the buitl XLA computation