jax-ml / jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
http://jax.readthedocs.io/
Apache License 2.0
30.59k stars 2.82k forks source link

Update the "CI" workflow to use the new self-hosted runners #25087

Open copybara-service[bot] opened 8 hours ago

copybara-service[bot] commented 8 hours ago

Update the "CI" workflow to use the new self-hosted runners

Rohanjames1997 commented 4 hours ago

Curious:

What is the rationality behind moving the CI workflow to self-hosted runners? What would be the effort to add an arm64 self-hosted runner as well as a target? cc: https://github.com/jax-ml/jax/discussions/25062