Tests

Merge pull request #1209 from AI-Hypercomputer:msingh-d #612

Job	Run time
prelim	1s
gpu_image / Build and upload image (a100-40gb-4)	40s
tpu_image / Build and upload image (v4-8)	3m 15s
gpu_integration_tests / run	8m 12s
gpu_unit_tests / run	2m 57s
tpu_integration_tests / run	6m 13s
tpu_unit_tests / run	16m 58s
Clean up	6s
	38m 22s

Provide feedback