Tests

Allow nsys profile saved to local directory #242

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

Provide feedback