xla

54101e76 - [PJRT] Change the name of `LOCAL_RANK` to fix conflicts (#4646)

[PJRT] Change the name of `LOCAL_RANK` to fix conflicts (#4646) * [PJRT] Change the name of `LOCAL_RANK` to fix conflicts * formatting * Update usage of local rank in GPU config