Support PyTorch/XLA PJRT for XLAStrategy #17350
Description
Description & Motivation
Add support for PJRT with the release of PyTorch/XLA 2.0.
PJRT provides several advantages over previous runtime XRT, and it also simplifies the workflow to start a TPU Pod job.
Pitch
No response
Alternatives
No response
Additional context
No response
cc @Borda @carmocca @justusschock @awaelchli @JackCaoG @steventk-g @Liyang90