refactor(API): improve usability #1319

Job	Run time
training_16GPU_4DP2TP2PP_FSP (t_cluster)	2m 3s
training_16GPU_4DP2TP2PP_MTP (t_cluster)	2m 36s
training_16GPU_4DP2TP2PP_MSP (t_cluster)	1m 47s
training_4GPU	2m 5s
training_8GPU_4DP2PP (t_cluster)	1m 45s
training_8GPU_4DP2PP_ZB	15m 17s
training_8GPU_4DP2TP (t_cluster)	1m 48s
training_8GPU_4DP2TPSP (t_cluster)	2m 1s
training_8GPU_ISP	2m 14s
training_8GPU_ISP_CKPT	2m 21s
training_llama2 (t_cluster)	1m 52s
	35m 49s

Provide feedback