Tarry Singh 0d5389ceda 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
..
cpu	Add Bazel build rules and runtime implementation for AWS Neuron/Trainium/Inferentia support.	2023-08-18 17:11:27 +00:00
cuda	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
neuron	Add Bazel build rules and runtime implementation for AWS Neuron/Trainium/Inferentia support.	2023-08-18 17:11:27 +00:00
rocm	Add Bazel build rules and runtime implementation for AWS Neuron/Trainium/Inferentia support.	2023-08-18 17:11:27 +00:00
tpu	Add Bazel build rules and runtime implementation for AWS Neuron/Trainium/Inferentia support.	2023-08-18 17:11:27 +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
runtimes.zig	Add Bazel build rules and runtime implementation for AWS Neuron/Trainium/Inferentia support.	2023-08-18 17:11:27 +00:00