mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-13 02:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/build
History
Mathew Odden ec4b8ee1ed Fixes for 0.5.0 build ported to rocm-main 
(cherry picked from commit c23a81461192a2b6da3d364076a261714d2dc64f)
2025-03-25 17:51:30 -05:00
..
rocm
Fixes for 0.5.0 build ported to rocm-main
2025-03-25 17:51:30 -05:00
tools
Add functionality to allow promoting RC wheels during release
2025-03-04 14:21:12 -08:00
BUILD.bazel
Add NVIDIA wheel requirements only for Linux builds.
2025-03-11 12:33:54 -07:00
build.py
Merge pull request #271 from ROCm/ci-upstream-sync-142_1
2025-03-11 14:10:03 -05: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
gpu-test-requirements.txt
Add NVIDIA wheel requirements only for Linux builds.
2025-03-11 12:33:54 -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
Upgrade ml-dtypes version in py3.10-py3.13 hermetic python lock files.
2025-03-11 14:41:34 -07:00
requirements_lock_3_11.txt
Upgrade ml-dtypes version in py3.10-py3.13 hermetic python lock files.
2025-03-11 14:41:34 -07:00
requirements_lock_3_12.txt
Upgrade ml-dtypes version in py3.10-py3.13 hermetic python lock files.
2025-03-11 14:41:34 -07:00
requirements_lock_3_13_ft.txt
Add NVIDIA wheel requirements only for Linux builds.
2025-03-11 12:33:54 -07:00
requirements_lock_3_13.txt
Upgrade ml-dtypes version in py3.10-py3.13 hermetic python lock files.
2025-03-11 14:41:34 -07:00
requirements.in
Update :build_jaxlib flag to control whether we should add py_import dependencies to the test targets.
2025-03-11 08:31:43 -07:00
test-requirements.txt
Pin numpy~=2.1.0 in workflow file instead of test-requirements.txt
2025-03-17 08:59:06 -07:00