Ruturaj Vaidya dceb5310fe

[ROCm] Implement RNN support (#217)

2025-02-07 11:08:12 -06:00

..

build_gpu_kernels_wheel.py

[ROCm] Implement RNN support (#217)

2025-02-07 11:08:12 -06:00

build_gpu_plugin_wheel.py

Replace plat-name with plat_name.

2024-09-18 15:17:49 +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

Add config option to log or fatal when jax.Arrays are GCed.

2024-10-17 12:23:16 -07:00

BUILD.bazel

Set PYTHONWARNINGS=error in bazel tests.

2024-09-24 12:30:11 -07: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