Add DeepSWE train script. #4189

Job	Run time
Build tunix package / Build Tunix wheel	10s
tunix_cpu_unit_tests / run	1m 24s
tunix_tpu_unit_tests / run_prod	1s
tunix_tpu_unit_tests / run_dev	2m 21s
Notify failed build	3s
	3m 59s

Provide feedback