PyTorch Lightning uses deprecated torch_xla function, causing compatibility issues with latest torch_xla versions

[aahila-aws]

Bug description

PyTorch Lightning uses deprecated torch_xla function, causing compatibility issues with latest torch_xla versions.
In this file we see that PyTorch Lightning using an xla function that has been deprecated in the latest versions of torch_xla.

What version are you seeing the problem on?

master

How to reproduce the bug

No response

Error messages and logs

# Error messages and logs here please

AttributeError: module 'torch_xla.runtime' has no attribute 'using_pjrt'

Environment

Current environment

#- PyTorch Lightning Version (e.g., 2.4.0):
#- PyTorch Version (e.g., 2.4):
#- Python version (e.g., 3.12):
#- OS (e.g., Linux):
#- CUDA/cuDNN version:
#- GPU models and configuration:
#- How you installed Lightning(`conda`, `pip`, source):

More info

No response