This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
mirror of
https://github.com/ROCm/jax.git
synced
2025-04-16 11:56:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jaxlib
/
tools
History
David Dunleavy
1a19d5594a
Update all uses of
@tsl//third_party
to
@xla//third_party
...
PiperOrigin-RevId: 733495240
2025-03-04 15:55:23 -08:00
..
build_gpu_kernels_wheel.py
Refactor JAX wheel build rules to control the wheel filename and maintain reproducible wheel content and filename results.
2025-02-05 10:01:23 -08:00
build_gpu_plugin_wheel.py
Refactor JAX wheel build rules to control the wheel filename and maintain reproducible wheel content and filename results.
2025-02-05 10:01:23 -08:00
build_utils.py
Update
jax_wheel
target to produce both wheel and source distribution files.
2025-02-27 07:41:13 -08: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 version check to jaxlib plugin imports.
2025-02-27 11:52:17 -08:00
BUILD.bazel
Update all uses of
@tsl//third_party
to
@xla//third_party
2025-03-04 15:55:23 -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
