xla

f01cdb60 - update doc to use PJRT_DEVICE=CUDA instead of PJRT_DEVICE=GPU (#5754)

update doc to use PJRT_DEVICE=CUDA instead of PJRT_DEVICE=GPU (#5754) * update doc to use PJRT_DEVICE=CUDA instead of PJRT_DEVICE=GPU * add warning message. * fix comment and test failure. * skip dynamic shape model test on cuda.