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

..

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

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

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

dependabot.yml

Update dependabot.yml

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