Cherry pick of #2324 #2336 #2337 into release-0.17 branch #632

Job	Run time
e2e (v1.27.11, long-running-resume,from-volume-resume,median-stop)	17m 3s
e2e (v1.27.11, grid,bayesian-optimization,tpe,multivariate-tpe,cma-es,hyperband)	21m 38s
e2e (v1.27.11, file-metrics-collector,pytorchjob-mnist)	12m 21s
e2e (v1.27.11, median-stop-with-json-format,file-metrics-collector-with-json-format)	12m 0s
e2e (v1.28.7, long-running-resume,from-volume-resume,median-stop)	15m 18s
e2e (v1.28.7, grid,bayesian-optimization,tpe,multivariate-tpe,cma-es,hyperband)	19m 28s
e2e (v1.28.7, file-metrics-collector,pytorchjob-mnist)	14m 23s
e2e (v1.28.7, median-stop-with-json-format,file-metrics-collector-with-json-format)	16m 6s
e2e (v1.29.2, long-running-resume,from-volume-resume,median-stop)	17m 3s
e2e (v1.29.2, grid,bayesian-optimization,tpe,multivariate-tpe,cma-es,hyperband)	19m 10s
e2e (v1.29.2, file-metrics-collector,pytorchjob-mnist)	17m 17s
e2e (v1.29.2, median-stop-with-json-format,file-metrics-collector-with-json-format)	13m 10s
	3h 14m 57s

Provide feedback