Add DeepSWE train script. #4234

Job	Run time
Build tunix package / Build Tunix wheel	12s
tunix_cpu_unit_tests / run	1m 21s
tunix_tpu_unit_tests / run_dev	24m 26s
tunix_tpu_unit_tests / run_prod	13m 5s
Notify failed build	2s
	39m 6s

Provide feedback