mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/build
History
vfdev-5 5a340a9781 Disable second order vjp tests in RunStateHypothesisTest.test_vjp if JAX_SKIP_SLOW_TESTS=true 
Description:
- Disable second order vjp tests in RunStateHypothesisTest.test_vjp if JAX_SKIP_SLOW_TESTS=true to reduce the test execution time
  - especially for TSAN CI job where this test takes ~700 seconds to pass with the recent 3.13 cpython
- Removed optional deps for 3.14
2025-04-08 21:02:55 +00:00
..
rocm
Fix auditwheel
2025-03-25 14:49:21 +00:00
tools
Fix problem finding clang++ when building JAX via build.py on windows.
2025-04-03 15:00:29 +01:00
BUILD.bazel
Add NVIDIA wheel requirements only for Linux builds.
2025-03-11 12:33:54 -07:00
build.py
Restrict the regex for copying the wheels.
2025-04-03 08:06:37 -07: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
Update the minimum supported CuDNN version to 9.8 (previously 9.1).
2025-03-27 09:54:00 -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
Update the minimum supported CuDNN version to 9.8 (previously 9.1).
2025-03-27 09:54:00 -07:00
requirements_lock_3_11.txt
Update the minimum supported CuDNN version to 9.8 (previously 9.1).
2025-03-27 09:54:00 -07:00
requirements_lock_3_12.txt
Update the minimum supported CuDNN version to 9.8 (previously 9.1).
2025-03-27 09:54:00 -07:00
requirements_lock_3_13_ft.txt
Updates for 3.14
2025-04-02 08:32:25 +00:00
requirements_lock_3_13.txt
Update the minimum supported CuDNN version to 9.8 (previously 9.1).
2025-03-27 09:54:00 -07:00
requirements_lock_3_14_ft.txt
Disable second order vjp tests in RunStateHypothesisTest.test_vjp if JAX_SKIP_SLOW_TESTS=true
2025-04-08 21:02:55 +00: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