Ruturaj4 fe68eb8b25 [ROCm] Implement RNN support

2025-01-14 19:04:49 -06:00

..

build_gpu_kernels_wheel.py

[ROCm] Implement RNN support

2025-01-14 19:04:49 -06:00

build_gpu_plugin_wheel.py

Add a jax_wheel Bazel rule to build jax pip packages

2024-10-24 14:20:46 -07: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

[mosaic_gpu] Include Mosaic GPU dialect fiels into jaxlib

2024-12-23 13:46:25 +00:00

BUILD.bazel

Use common set of build options when building jaxlib+plugin artifacts together

2024-12-19 14:29:24 -08:00

gpu_version_script.lds

Add basic PyTorch integration for Mosaic GPU

2024-09-18 12:55:23 +00:00

LICENSE.txt

Remove DUCC FFT from jaxlib

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