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

History

Nitin Srinivasan c8ccd7570a Add functionality that let us do a "jax" only release

Introduces a new `download-jax-only-from-gcs` variable to the workflow configs. When set to 1, the test workflows will only download and install the `jax` wheel. Other artifacts such as the latest releases of `jaxlib` and the CUDA plugin dependencies will be downloaded and installed from PyPI.

PiperOrigin-RevId: 740430538

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

envs

Add TPU test jobs to the new CI continuous and nightly/release test workflows

2025-03-11 11:42:21 -07:00

utilities

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

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

build_artifacts.sh

Add functionality to allow promoting RC wheels during release

2025-03-04 14:21:12 -08: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

Add new CI script to run Bazel GPU (non-RBE) jobs

2024-11-26 18:42:27 -08: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

Replace "gpu" with "cuda" to be specific about the type of gpu tests that are running

2025-01-22 09:40:11 -08: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.