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

This commit adds the CI script needed for running Bazel GPU (non-RBE) tests. These run two Bazel commands: Single accelerator tests with one GPU a piece and multi-accelerator tests with all GPUs PiperOrigin-RevId: 700523594
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.