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 615219b1f6 Remove tensorstore dependency from //jax/experimental/array_serialization:serialization in OSS (see https://github.com/google/tensorstore/issues/218) 
Disable serialization_test in OSS.

PiperOrigin-RevId: 731463136
2025-02-26 14:47:16 -08:00
..
rocm
Merge pull request #25810 from ROCm:gh-9948-add-gpu-ci-upstream
2025-02-07 09:51:59 -08:00
tools
Upgrade to Bazel 7.4.1
2025-02-26 05:33:24 -08: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
Remove tensorstore dependency from //jax/experimental/array_serialization:serialization in OSS (see https://github.com/google/tensorstore/issues/218)
2025-02-26 14:47:16 -08:00
requirements_lock_3_11.txt
Remove tensorstore dependency from //jax/experimental/array_serialization:serialization in OSS (see https://github.com/google/tensorstore/issues/218)
2025-02-26 14:47:16 -08:00
requirements_lock_3_12.txt
Remove tensorstore dependency from //jax/experimental/array_serialization:serialization in OSS (see https://github.com/google/tensorstore/issues/218)
2025-02-26 14:47:16 -08:00
requirements_lock_3_13_ft.txt
Remove tensorstore dependency from //jax/experimental/array_serialization:serialization in OSS (see https://github.com/google/tensorstore/issues/218)
2025-02-26 14:47:16 -08:00
requirements_lock_3_13.txt
Remove tensorstore dependency from //jax/experimental/array_serialization:serialization in OSS (see https://github.com/google/tensorstore/issues/218)
2025-02-26 14:47:16 -08:00
requirements.in
Reverts 72f9a493589a1046e6927a5f16d7dc71df530743
2024-10-31 10:05:22 -07:00
test-requirements.txt
Remove tensorstore dependency from //jax/experimental/array_serialization:serialization in OSS (see https://github.com/google/tensorstore/issues/218)
2025-02-26 14:47:16 -08:00