mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/build
History
jax authors e64650e2ba Add --config=cuda_libraries_from_stubs in the end of all additional bazel options for CUDA wheels. 
Build.py shouldn't be used for building the wheels with real CUDA libraries in the dependencies. This change prevents overriding the default configuration.

PiperOrigin-RevId: 725326252
2025-02-10 13:24:47 -08:00
..
rocm
Merge pull request #25810 from ROCm:gh-9948-add-gpu-ci-upstream
2025-02-07 09:51:59 -08:00
tools
Fix a missing "f" on an f-string.
2025-02-07 12:53:24 -05:00
BUILD.bazel
Use hermetic Python in JAX, see "Managing hermetic Python" in developer.md for details
2024-05-15 18:20:56 -07:00
build.py
Add --config=cuda_libraries_from_stubs in the end of all additional bazel options for CUDA wheels.
2025-02-10 13:24:47 -08:00
collect-profile-requirements.txt
Cloud TPU CI: make sure we update test deps and upgrade protobuf version
2023-07-19 16:17:47 -07:00
parallel_accelerator_execute.sh
Add script for parallel accelerator testing under Bazel.
2022-07-06 10:58:04 -04:00
requirements_lock_3_10.txt
Reverts 72f9a493589a1046e6927a5f16d7dc71df530743
2024-10-31 10:05:22 -07:00
requirements_lock_3_11.txt
Reverts 72f9a493589a1046e6927a5f16d7dc71df530743
2024-10-31 10:05:22 -07:00
requirements_lock_3_12.txt
Reverts 72f9a493589a1046e6927a5f16d7dc71df530743
2024-10-31 10:05:22 -07:00
requirements_lock_3_13_ft.txt
Added 3.13 ft requirements lock file and updated WORKSPACE
2025-01-08 22:47:29 +01:00
requirements_lock_3_13.txt
Reverts 72f9a493589a1046e6927a5f16d7dc71df530743
2024-10-31 10:05:22 -07:00
requirements.in
Reverts 72f9a493589a1046e6927a5f16d7dc71df530743
2024-10-31 10:05:22 -07:00
test-requirements.txt
Remove portpicker for free threaded python 3.13t in test-requirements.txt
2025-02-03 13:30:01 -08:00