make trace writing a configurable option #4218

Job	Run time
Build tunix package / Build Tunix wheel	12s
tunix_cpu_unit_tests / run	1m 12s
tunix_tpu_unit_tests / run_prod	12m 22s
tunix_tpu_unit_tests / run_dev	7m 45s
Notify failed build	5s
	21m 36s

Provide feedback