mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00

History

jax authors 40a3d0c78d Create the test targets for the wheel size verification.

Add the tests to the Bazel presubmit RBE jobs (except `arm64`/`aarch64` jobs that use RBE cross-compilation).

PiperOrigin-RevId: 742724458

2025-04-01 09:11:56 -07:00

envs

Update the Windows docker image to ltsc2022

2025-03-26 15:20:09 -07:00

utilities

Add functionality that let us do a "jax" only release

2025-03-25 11:48:29 -07:00

build_artifacts.sh

build/build.py changes: copy the wheels created by the new build wheel targets into the path specified by --output_path.

2025-03-26 10:56:19 -07:00

README.md

Add new CI scripts for running Bazel CPU presubmits

2024-11-12 20:00:35 -08:00

run_bazel_test_cpu_rbe.sh

Create the test targets for the wheel size verification.

2025-04-01 09:11:56 -07:00

run_bazel_test_cuda_non_rbe.sh

Update the calculation for num_processes and num_test_jobs that are used in CUDA test jobs

2025-02-27 08:11:51 -08:00

run_bazel_test_cuda_rbe.sh

Create the test targets for the wheel size verification.

2025-04-01 09:11:56 -07:00

run_pytest_cpu.sh

Add a presubmit check to test against oldest supported numpy

2025-03-19 13:48:49 -07:00

run_pytest_cuda.sh

Print the list of installed packages before running pytests

2025-02-28 11:46:12 -08:00

run_pytest_tpu.sh

Ignore Pallas TPU tests when testing with the oldest supported libtpu

2025-03-12 05:20:42 -07:00

README.md

JAX continuous integration

Warning

This folder is still under construction. It is part of an ongoing effort to improve the structure of CI and build related files within the JAX repo. This warning will be removed when the contents of this directory are stable and appropriate documentation around its usage is in place.