jax
f07243a7 - Default JAX_CPU_COLLECTIVES_IMPLEMENTATION to 'gloo'.

Commit

330 days ago

Default JAX_CPU_COLLECTIVES_IMPLEMENTATION to 'gloo'. This enables CPU collectives by default, making multi-process CPU communication work without extra configuration. PiperOrigin-RevId: 724076284

References

#26264 - Default JAX_CPU_COLLECTIVES_IMPLEMENTATION to 'gloo'.

#227 - CI: 02/10/25 upstream sync

Author

skye

Committer

Google-ML-Automation

Parents

4b86ff22

jax f07243a7 - Default JAX_CPU_COLLECTIVES_IMPLEMENTATION to 'gloo'.

jax
f07243a7 - Default JAX_CPU_COLLECTIVES_IMPLEMENTATION to 'gloo'.