Skip to content

update doc to use PJRT_DEVICE=CUDA instead of PJRT_DEVICE=GPU#5754

Merged
vanbasten23 merged 4 commits intomasterfrom replaceGpuWithCudaNov 4, 2023