Ruturaj4 332435e028 [ROCM] make mosaic dependency cuda specific

2024-07-02 11:05:42 -05:00

..

build_gpu_kernels_wheel.py

libdevice.10.bc is removed from JAX wheels bundle.

2024-06-27 08:35:59 -07:00

build_gpu_plugin_wheel.py

[ROCm] Bring-up pjrt support

2024-06-17 16:49:22 +00:00

build_wheel_test.py

Add a bazel test that verifies that the jaxlib wheel builds.

2024-01-11 23:22:17 +00:00

build_wheel.py

libdevice.10.bc is removed from JAX wheels bundle.

2024-06-27 08:35:59 -07:00

BUILD.bazel

[ROCM] make mosaic dependency cuda specific

2024-07-02 11:05:42 -05:00

LICENSE.txt

Remove DUCC FFT from jaxlib

2024-05-28 21:12:23 -07:00