refactor(API): improve usability #1387

Job	Run time
training_4GPU	2m 10s
training_8GPU_ISP	1m 45s
training_16GPU_4DP2TP2PP_FSP (t_cluster)	1m 58s
training_16GPU_4DP2TP2PP_MSP (t_cluster)	15m 16s
training_16GPU_4DP2TP2PP_MTP (t_cluster)	26s
training_8GPU_4DP2PP (t_cluster)	4m 32s
training_8GPU_4DP2PP_ZB	4m 27s
training_8GPU_4DP2TPSP (t_cluster)	2m 8s
training_8GPU_4DP2TP (t_cluster)	2m 15s
training_8GPU_ISP_CKPT	2m 38s
training_llama2 (t_cluster)	1m 55s
	39m 30s

Provide feedback