Disable for_loop_test on TPU v5p.

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.47k stars 2.8k forks source link

Closed copybara-service[bot] closed 1 day ago

copybara-service[bot] commented 4 days ago

Disable for_loop_test on TPU v5p.

This test is failing in CI.