Tests

Merge pull request #1226 from AI-Hypercomputer:patemotter_tol_fix #711

Job	Run time
prelim	22s
gpu_image / Build and upload image (a100-40gb-4)	8s
tpu_image / Build and upload image (v4-8)	3m 13s
gpu_integration_tests / run	7m 39s
gpu_unit_tests / run	2m 18s
tpu_unit_tests / run	16m 12s
tpu_integration_tests / run	6m 36s
Clean up	5s
	36m 33s

Provide feedback