xla

939f8961 - Parallel tests (#5066)

Parallel tests (#5066) * Parallel tests (#4960) (#5030) * Improve coverage config * Rearrange tests to speed them up. * Run TPU and GPU tests serially. * Automate sharding * Suppress logging to observe what happens with tests.