jax
6ef54da7 - Lower JAX_TEST_NUM_THREADS to 32 from 64 for TPU tests.

Commit

24 days ago

Lower JAX_TEST_NUM_THREADS to 32 from 64 for TPU tests. This is an attempt to reduce flaky test timeouts. PiperOrigin-RevId: 964700756

References

Author

hawkinsp

Committer

Google-ML-Automation

Parents