diff --git a/test/kubernetes/benchmarks/pytorch.go b/test/kubernetes/benchmarks/pytorch.go
index 4d382fcac..e956c13eb 100644
--- a/test/kubernetes/benchmarks/pytorch.go
+++ b/test/kubernetes/benchmarks/pytorch.go
@@ -144,6 +144,15 @@ var (
 			test:   eval,
 		},
 	}
+
+	// AllTests is a map of test names to the tests.
+	AllTests = map[string][]pytorchTest{
+		"FastNLPBert":       FastNLPBert,
+		"BigBird":           BigBird,
+		"SpeechTransformer": SpeechTransformer,
+		"LearningToPaint":   LearningToPaint,
+		"MobileNetV2":       MobileNetV2,
+	}
 )
 
 // Name returns the name of the test with the argument parameters included. It is formatted so