Tests

fixing types and updating main in maxengine and inference_microbenchm… #601

Job	Run time
prelim	1s
tpu_image / Build and upload image (v4-8)	3m 7s
gpu_image / Build and upload image (a100-40gb-4)	40s
gpu_unit_tests / run	3m 1s
gpu_integration_tests / run	7m 49s
tpu_integration_tests / run	6m 32s
tpu_unit_tests / run	16m 10s
Clean up	5s
	37m 25s

Provide feedback