Tests

Merge pull request #1132 from AI-Hypercomputer:small_gpu_build_optimm… #50

Job	Run time
prelim	8s
gpu_image / Build and upload image (a100-40gb-4)	14m 1s
tpu_image / Build and upload image (v4-8)	6s
tpu_unit_tests / run	13m 4s
tpu_integration_tests / run	6m 4s
gpu_unit_tests / run	4m 55s
gpu_integration_tests / run	7m 40s
Clean up	8s
	46m 6s

Provide feedback