Radix/runtimes
2025-04-10 14:47:16 +00:00
..
common runtimes/cuda: sandbox CUDA dependencies by removing them from the leaf binary, sandboxing the dependency graph, marking dlopen direct dependencies as NEEDED, setting RPATH to the sandbox, loading the PJRT plugin from the sandbox, and enabling weak CUDA symbols without direct linking. 2025-03-26 11:18:29 +00:00
cpu runtimes/cpu: sandbox CPU PJRT plugin, simplifying as there are no additional NEEDED dependencies. 2025-04-03 11:57:46 +00:00
cuda runtimes/cuda: sandbox CUDA dependencies by removing them from the leaf binary, sandboxing the dependency graph, marking dlopen direct dependencies as NEEDED, setting RPATH to the sandbox, loading the PJRT plugin from the sandbox, and enabling weak CUDA symbols without direct linking. 2025-03-26 11:18:29 +00:00
neuron Update neuron runtime BUILD.bazel to use Bazel manual tag and S3 cache integration. 2025-02-03 14:03:33 +00:00
rocm workspace: bump xla to version 20250710.0-22ea002 2025-03-04 17:12:34 +00:00
tpu runtimes/tpu: sandbox TPU PJRT plugin; no external dependencies. 2025-04-10 14:47:16 +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