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

Job	Run time
e2e (v1.27.11, long-running-resume,from-volume-resume,median-stop)	21m 26s
e2e (v1.27.11, grid,bayesian-optimization,tpe,multivariate-tpe,cma-es,hyperband)	25m 39s
e2e (v1.27.11, file-metrics-collector,pytorchjob-mnist)	12m 56s
e2e (v1.27.11, median-stop-with-json-format,file-metrics-collector-with-json-format)	22m 55s
e2e (v1.28.7, long-running-resume,from-volume-resume,median-stop)	18m 27s
e2e (v1.28.7, grid,bayesian-optimization,tpe,multivariate-tpe,cma-es,hyperband)	21m 27s
e2e (v1.28.7, file-metrics-collector,pytorchjob-mnist)	18m 52s
e2e (v1.28.7, median-stop-with-json-format,file-metrics-collector-with-json-format)	12m 33s
e2e (v1.29.2, long-running-resume,from-volume-resume,median-stop)	20m 47s
e2e (v1.29.2, grid,bayesian-optimization,tpe,multivariate-tpe,cma-es,hyperband)	19m 2s
e2e (v1.29.2, file-metrics-collector,pytorchjob-mnist)	13m 31s
e2e (v1.29.2, median-stop-with-json-format,file-metrics-collector-with-json-format)	19m 33s
	3h 47m 8s

Provide feedback