xla

[PJRT] Defer default device selection until we need to know the runtime

#4630

Merged