xla

eb37dfdf - Remove torchax from torchxla repo. (#9672)

Remove torchax from torchxla repo. (#9672) It's now in a separate repo at google/torchax Optional dependency on torchax is handled the same way as JAX, (via pip dependency)