mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/build
History
Nitin Srinivasan 031614c22b Pin numpy~=2.1.0 in workflow file instead of test-requirements.txt 
PiperOrigin-RevId: 737632771
2025-03-17 08:59:06 -07:00
..
rocm
Merge pull request #25810 from ROCm:gh-9948-add-gpu-ci-upstream
2025-02-07 09:51:59 -08: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
Add functionality to allow promoting RC wheels during release
2025-03-04 14:21:12 -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
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