Disable tests for int4 support on non-TPU platforms.
An upcoming XLA change will reject programs containing int4 on CPU and GPU, because the XLA support is buggy and incomplete. When the XLA supports this we can reenable these tests.
Issue https://github.com/google/jax/issues/17672
PiperOrigin-RevId: 570042917
