jax authors 727d0367a4 Update --config=cuda to add direct dependencies on CUDA libraries both for bazel build and bazel test phases. ...

With this configuration the same cache is used both for `bazel build` and `bazel test` commands (provided the same target is specified).

Add `--config=no_cuda_libs` for building targets with CUDA libraries from stubs.

PiperOrigin-RevId: 720334587

2025-01-27 15:46:17 -08:00

..

rocm

Merge pull request #25917 from ROCm:ci_fix_multi_gpu_test_logic-upstream

2025-01-16 02:45:54 -08:00

tools

Disable avxvnniint8 when building with Clang version < 19, or GCC < 13.

2025-01-06 07:06:09 -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

Update --config=cuda to add direct dependencies on CUDA libraries both for bazel build and bazel test phases.

2025-01-27 15:46:17 -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

Set minimum supported Python version to 3.10 for matplotlib.

2024-11-01 12:34:28 -07:00