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

..

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

Build JAX wheels instead of installing it from the source repository

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

dependabot.yml

Update dependabot.yml

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