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-17 12:26:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jaxlib
/
tools
History
Kanglan Tang
af28595909
Add a jax_wheel Bazel rule to build jax pip packages
...
PiperOrigin-RevId: 689514531
2024-10-24 14:20:46 -07:00
..
build_gpu_kernels_wheel.py
Add a jax_wheel Bazel rule to build jax pip packages
2024-10-24 14:20:46 -07: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
Add a jax_wheel Bazel rule to build jax pip packages
2024-10-24 14:20:46 -07:00
BUILD.bazel
Add a jax_wheel Bazel rule to build jax pip packages
2024-10-24 14:20:46 -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
