Tarry Singh 455bb3877f runtimes/cuda: obtain NCCL from the pip package, matching XLA behavior.		2023-09-20 17:41:44 +00:00
..
BUILD.bazel	Update CUDA runtime sandboxing and dynamic symbol renaming, switch to pre‑built jax‑cuda‑pjrt plugin, and bump CUDA to 12.6.2 and cuDNN to 9.5.1.	2023-09-14 13:28:25 +00:00
cuda.bzl	runtimes/cuda: obtain NCCL from the pip package, matching XLA behavior.	2023-09-20 17:41:44 +00:00
cuda.redistrib_12.6.3.json	Update CUDA runtime sandboxing and dynamic symbol renaming, switch to pre‑built jax‑cuda‑pjrt plugin, and bump CUDA to 12.6.2 and cuDNN to 9.5.1.	2023-09-14 13:28:25 +00:00
cuda.zig	Update CUDA runtime sandboxing and dynamic symbol renaming, switch to pre‑built jax‑cuda‑pjrt plugin, and bump CUDA to 12.6.2 and cuDNN to 9.5.1.	2023-09-14 13:28:25 +00:00
cudnn.redistrib_9.5.1.json	Update CUDA runtime sandboxing and dynamic symbol renaming, switch to pre‑built jax‑cuda‑pjrt plugin, and bump CUDA to 12.6.2 and cuDNN to 9.5.1.	2023-09-14 13:28:25 +00:00
libpjrt_cuda.BUILD.bazel	runtimes/cuda: obtain NCCL from the pip package, matching XLA behavior.	2023-09-20 17:41:44 +00:00
zmlxcuda.c	Update CUDA runtime sandboxing and dynamic symbol renaming, switch to pre‑built jax‑cuda‑pjrt plugin, and bump CUDA to 12.6.2 and cuDNN to 9.5.1.	2023-09-14 13:28:25 +00:00