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

History

Nitin Srinivasan 623865fe95 Build JAX wheels instead of installing it from the source repository

This change allows us to get rid of extra env vars which used to control whether to install `jax` at head. Now, `jax` will be be built and consumed in the same way as the other wheels in the continuous jobs.

PiperOrigin-RevId: 734123590

2025-03-06 06:48:16 -08:00

envs

Build JAX wheels instead of installing it from the source repository

2025-03-06 06:48:16 -08:00

utilities

Build JAX wheels instead of installing it from the source repository

2025-03-06 06:48:16 -08: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

Print the list of installed packages before running pytests

2025-02-28 11:46:12 -08: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

Print the list of installed packages before running pytests

2025-02-28 11:46:12 -08: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.