diff --git a/.kokoro/github/ubuntu/gpu/jax/continuous.cfg b/.kokoro/github/ubuntu/gpu/jax/continuous.cfg
index 90b62b22c9..c16a64fae0 100644
--- a/.kokoro/github/ubuntu/gpu/jax/continuous.cfg
+++ b/.kokoro/github/ubuntu/gpu/jax/continuous.cfg
@@ -31,4 +31,4 @@ before_action {
 }
 
 # Set timeout lower from default 180 mins
-timeout_mins: 120
\ No newline at end of file
+timeout_mins: 150
\ No newline at end of file
diff --git a/.kokoro/github/ubuntu/gpu/jax/presubmit.cfg b/.kokoro/github/ubuntu/gpu/jax/presubmit.cfg
index 90b62b22c9..c16a64fae0 100644
--- a/.kokoro/github/ubuntu/gpu/jax/presubmit.cfg
+++ b/.kokoro/github/ubuntu/gpu/jax/presubmit.cfg
@@ -31,4 +31,4 @@ before_action {
 }
 
 # Set timeout lower from default 180 mins
-timeout_mins: 120
\ No newline at end of file
+timeout_mins: 150
\ No newline at end of file
diff --git a/.kokoro/github/ubuntu/gpu/tensorflow/continuous.cfg b/.kokoro/github/ubuntu/gpu/tensorflow/continuous.cfg
index a5ddd0db8c..f0ba8e612b 100644
--- a/.kokoro/github/ubuntu/gpu/tensorflow/continuous.cfg
+++ b/.kokoro/github/ubuntu/gpu/tensorflow/continuous.cfg
@@ -31,4 +31,4 @@ before_action {
 }
 
 # Set timeout lower from default 180 mins
-timeout_mins: 120
\ No newline at end of file
+timeout_mins: 150
\ No newline at end of file
diff --git a/.kokoro/github/ubuntu/gpu/tensorflow/presubmit.cfg b/.kokoro/github/ubuntu/gpu/tensorflow/presubmit.cfg
index a5ddd0db8c..f0ba8e612b 100644
--- a/.kokoro/github/ubuntu/gpu/tensorflow/presubmit.cfg
+++ b/.kokoro/github/ubuntu/gpu/tensorflow/presubmit.cfg
@@ -31,4 +31,4 @@ before_action {
 }
 
 # Set timeout lower from default 180 mins
-timeout_mins: 120
\ No newline at end of file
+timeout_mins: 150
\ No newline at end of file