Tarry Singh edc2ac26f8 Adjust ROCm runtime sandboxing to hook only the PJRT plugin and make hipblastlt bytecodes optional.		2024-01-26 13:02:23 +00:00
..
cpu	Fix async hangs by reworking the libxev epoll backend and using callBlocking for PJRT plugin loading, improving performance across async and runtime modules.	2024-01-16 14:13:45 +00:00
cuda	Fix async hangs by reworking the libxev epoll backend and using callBlocking for PJRT plugin loading, improving performance across async and runtime modules.	2024-01-16 14:13:45 +00:00
neuron	Fix async hangs by reworking the libxev epoll backend and using callBlocking for PJRT plugin loading, improving performance across async and runtime modules.	2024-01-16 14:13:45 +00:00
rocm	Adjust ROCm runtime sandboxing to hook only the PJRT plugin and make hipblastlt bytecodes optional.	2024-01-26 13:02:23 +00:00
tpu	Fix async hangs by reworking the libxev epoll backend and using callBlocking for PJRT plugin loading, improving performance across async and runtime modules.	2024-01-16 14:13:45 +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