diff --git a/paxml/pip_package/cloudbuild-release.yaml b/paxml/pip_package/cloudbuild-release.yaml
index 4ab8b3a61..d2ffcf355 100644
--- a/paxml/pip_package/cloudbuild-release.yaml
+++ b/paxml/pip_package/cloudbuild-release.yaml
@@ -24,6 +24,7 @@ substitutions:
 options:
     dynamic_substitutions: true
     substitution_option: 'ALLOW_LOOSE'
+    machineType: E2_HIGHCPU_8
 timeout: 5400s
 artifacts:
   objects:
diff --git a/paxml/pip_package/requirements.txt b/paxml/pip_package/requirements.txt
index bd6ff5a40..1f3bbd3e9 100644
--- a/paxml/pip_package/requirements.txt
+++ b/paxml/pip_package/requirements.txt
@@ -114,7 +114,7 @@ executing==1.2.0
     # via stack-data
 fastjsonschema==2.16.2
     # via nbformat
-fiddle @ git+https://github.com/google/fiddle
+fiddle==0.2.4
     # via -r praxis-requirements.in
 filelock==3.8.0
     # via