jax
de48ce2a - Merge pull request #26174 from skye:cpu_configs

Commit

340 days ago

Merge pull request #26174 from skye:cpu_configs PiperOrigin-RevId: 722199229

References

#26174 - Make `JAX_CPU_COLLECTIVES_IMPLEMENTATION` and `JAX_NUM_CPU_DEVICES` env vars

#219 - CI: 02/03/25 upstream sync

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading