diff --git a/pyproject.toml b/pyproject.toml
index d1ef06e..363638a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -38,6 +38,15 @@ dev = [
     "pylint>=2.6.0",
     "pyink",
 ]
+gke = [
+    "kubernetes",
+    "pyyaml",
+]
+test = [
+    "kubernetes",
+    "pyyaml",
+    "pytest",
+]
 
 [tool.pyink]
 # Formatting configuration to follow Google style-guide
diff --git a/requirements.txt b/requirements.txt
index ee86a9e..62ab9f5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,3 +5,5 @@ orbax-checkpoint
 uvicorn
 requests
 packaging
+kubernetes
+pyyaml