jax
66b90054 - Disable pjit ArrayPjitTest.test_device_put_grad test on TPU v5e

Commit

1 year ago

Disable pjit ArrayPjitTest.test_device_put_grad test on TPU v5e PiperOrigin-RevId: 704378732

References

Author

kanglant

Committer

Google-ML-Automation

Parents