xla

37bfcd88 - Parallel tests (#4960)

Parallel tests (#4960) * Improve coverage config * Rearrange tests to speed them up. * Run TPU and GPU tests serially. * Automate sharding