Add DeepSWE train script. #4233

Job	Run time
Build tunix package / Build Tunix wheel	10s
tunix_cpu_unit_tests / run	1m 17s
tunix_tpu_unit_tests / run_dev	25m 2s
tunix_tpu_unit_tests / run_prod	13m 18s
Notify failed build	4s
	39m 51s

Provide feedback