diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 6bac4a1e..83157128 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -18,7 +18,7 @@ jobs:
     - name: Setup Go
       uses: actions/setup-go@v2
       with:
-        go-version: 1.16
+        go-version: 1.17
 
     - uses: actions/cache@v2
       with:
diff --git a/pkg/apis/common/v1/openapi_generated.go b/pkg/apis/common/v1/openapi_generated.go
index 3a9cf462..d0af1676 100644
--- a/pkg/apis/common/v1/openapi_generated.go
+++ b/pkg/apis/common/v1/openapi_generated.go
@@ -1,3 +1,4 @@
+//go:build !ignore_autogenerated
 // +build !ignore_autogenerated
 
 // Copyright 2021 The Kubeflow Authors
@@ -480,9 +481,17 @@ func schema_pkg_apis_common_v1_ReplicaStatus(ref common.ReferenceCallback) commo
 							Format:      "int32",
 						},
 					},
+					"labelSelector": {
+						SchemaProps: spec.SchemaProps{
+							Description: "A label selector is a label query over a set of resources. The result of matchLabels and matchExpressions are ANDed. An empty label selector matches all objects. A null label selector matches no objects.",
+							Ref:         ref("k8s.io/apimachinery/pkg/apis/meta/v1.LabelSelector"),
+						},
+					},
 				},
 			},
 		},
+		Dependencies: []string{
+			"k8s.io/apimachinery/pkg/apis/meta/v1.LabelSelector"},
 	}
 }
 
diff --git a/pkg/apis/common/v1/types.go b/pkg/apis/common/v1/types.go
index 0b894b26..e4eb7eb2 100644
--- a/pkg/apis/common/v1/types.go
+++ b/pkg/apis/common/v1/types.go
@@ -61,6 +61,11 @@ type ReplicaStatus struct {
 
 	// The number of pods which reached phase Failed.
 	Failed int32 `json:"failed,omitempty"`
+
+	// A label selector is a label query over a set of resources. The result of matchLabels and
+	// matchExpressions are ANDed. An empty label selector matches all objects. A null
+	// label selector matches no objects.
+	LabelSelector *metav1.LabelSelector `json:"labelSelector,omitempty"`
 }
 
 // +k8s:openapi-gen=true
diff --git a/pkg/apis/common/v1/zz_generated.deepcopy.go b/pkg/apis/common/v1/zz_generated.deepcopy.go
index 3d76bf20..d80060be 100644
--- a/pkg/apis/common/v1/zz_generated.deepcopy.go
+++ b/pkg/apis/common/v1/zz_generated.deepcopy.go
@@ -1,3 +1,4 @@
+//go:build !ignore_autogenerated
 // +build !ignore_autogenerated
 
 // Copyright 2021 The Kubeflow Authors
@@ -21,6 +22,7 @@ package v1
 import (
 	corev1 "k8s.io/api/core/v1"
 	resource "k8s.io/apimachinery/pkg/api/resource"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
 // DeepCopyInto is an autogenerated deepcopy function, copying the receiver, writing into out. in must be non-nil.
@@ -61,7 +63,7 @@ func (in *JobStatus) DeepCopyInto(out *JobStatus) {
 			} else {
 				in, out := &val, &outVal
 				*out = new(ReplicaStatus)
-				**out = **in
+				(*in).DeepCopyInto(*out)
 			}
 			(*out)[key] = outVal
 		}
@@ -116,6 +118,11 @@ func (in *ReplicaSpec) DeepCopy() *ReplicaSpec {
 // DeepCopyInto is an autogenerated deepcopy function, copying the receiver, writing into out. in must be non-nil.
 func (in *ReplicaStatus) DeepCopyInto(out *ReplicaStatus) {
 	*out = *in
+	if in.LabelSelector != nil {
+		in, out := &in.LabelSelector, &out.LabelSelector
+		*out = new(metav1.LabelSelector)
+		(*in).DeepCopyInto(*out)
+	}
 	return
 }
 
diff --git a/pkg/apis/common/v1/zz_generated.defaults.go b/pkg/apis/common/v1/zz_generated.defaults.go
index 6c5c1db1..1c9796c7 100644
--- a/pkg/apis/common/v1/zz_generated.defaults.go
+++ b/pkg/apis/common/v1/zz_generated.defaults.go
@@ -1,3 +1,4 @@
+//go:build !ignore_autogenerated
 // +build !ignore_autogenerated
 
 // Copyright 2021 The Kubeflow Authors
diff --git a/test_job/apis/test_job/v1/openapi_generated.go b/test_job/apis/test_job/v1/openapi_generated.go
index b33a3497..183f36c1 100644
--- a/test_job/apis/test_job/v1/openapi_generated.go
+++ b/test_job/apis/test_job/v1/openapi_generated.go
@@ -1,3 +1,4 @@
+//go:build !ignore_autogenerated
 // +build !ignore_autogenerated
 
 // Copyright 2021 The Kubeflow Authors
diff --git a/test_job/apis/test_job/v1/zz_generated.deepcopy.go b/test_job/apis/test_job/v1/zz_generated.deepcopy.go
index 48452cce..7efeceb2 100644
--- a/test_job/apis/test_job/v1/zz_generated.deepcopy.go
+++ b/test_job/apis/test_job/v1/zz_generated.deepcopy.go
@@ -1,3 +1,4 @@
+//go:build !ignore_autogenerated
 // +build !ignore_autogenerated
 
 // Copyright 2021 The Kubeflow Authors
diff --git a/test_job/apis/test_job/v1/zz_generated.defaults.go b/test_job/apis/test_job/v1/zz_generated.defaults.go
index 243e07fe..9a0338a4 100644
--- a/test_job/apis/test_job/v1/zz_generated.defaults.go
+++ b/test_job/apis/test_job/v1/zz_generated.defaults.go
@@ -1,3 +1,4 @@
+//go:build !ignore_autogenerated
 // +build !ignore_autogenerated
 
 // Copyright 2021 The Kubeflow Authors