Nitin Srinivasan 93831bdde7 Download and use jax wheels from GCS bucket for nightly/release test workflows ...

Unlike continuous workflows, when testing nightly/release artifacts, we want to download and install the `jax` wheels found in the GCS bucket instead of installing it from HEAD.

It looks like `env` setting in the calling workflow isn't passed over to the called workflows so we define a new workflow input, `install-jax-current-commit`, to control the `jax` install behavior.

PiperOrigin-RevId: 726086522

2025-02-12 09:32:05 -08:00

..

ISSUE_TEMPLATE

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

workflows

Download and use jax wheels from GCS bucket for nightly/release test workflows

2025-02-12 09:32:05 -08:00

dependabot.yml

Update dependabot.yml

2021-08-19 20:21:16 +02:00