xla

4cb838a2 - Make XLAShardedTensor use default __torch_function__