Replace `using_pjrt()` xla runtime `device_type()` check with in xla.py for `torch-xla>=2.5`#20442

lantiga merged 6 commits intoLightning-AI:masterLightning-AI/pytorch-lightning:masterfrom

byi8220:patch-1byi8220/pytorch-lightning:patch-1Copy head branch name to clipboard