jax

7cf236f8 - [jax2tf] Clean up test_qr and disable (temporarily) some tests for shirt_right (#4044)

[jax2tf] Clean up test_qr and disable (temporarily) some tests for shirt_right (#4044) on TPU This is a follow-up from #4032 that enabled some more integer types for TPU for JAX. There are still some silent errors to investigate for shift_right on TPU.