diff --git a/mjx/cuda_requirements.txt b/mjx/cuda_requirements.txt
index 0851eb61..afc63fe5 100644
--- a/mjx/cuda_requirements.txt
+++ b/mjx/cuda_requirements.txt
@@ -16,6 +16,8 @@ jax-cuda12-pjrt==0.5.3; python_version >= '3.10' \
 jax-cuda12-pjrt==0.4.30; python_version == '3.9' \
     --hash=sha256:895d0198ad99638fcaf976c47592e2a543eef79ea15fabd24a402d055390c328 \
     --hash=sha256:c36fb1e0c236563bf3a87e70f4d1ab28a31d7cf5d722c9ede30c4172116e8bcb
-warp-lang==1.9.0 \
-    --hash=sha256:23165d3291eeecc5ac47b9a3de0b93d34b4c921414e5a761ab96d68861efa3cb \
-    --hash=sha256:7ea8057e5d6fdb9b0885d725b51954c5fc6dac62d6ac007828148b1cb13ee615
+warp-lang==1.9.1 \
+    --hash=sha256:3ce61319a268f38d135f4a26c4e75c29488a88cf3af273896e1fb2bf174131ec \
+    --hash=sha256:2211e0cd9332ca44193801285cd1293c0b7580932da8da32e18d70b832adac12 \
+    --hash=sha256:e58d0503b34d484d4699077ac4161a342d18714e5eb9bf63ede9d5df2188ecf0 \
+    --hash=sha256:c971ed996004f729ceaf38b660831e95c837758df8ce6413a5d87d76652b0f88
diff --git a/mjx/pyproject.toml b/mjx/pyproject.toml
index b0254a39..2dc89e2b 100644
--- a/mjx/pyproject.toml
+++ b/mjx/pyproject.toml
@@ -36,7 +36,7 @@ dependencies = [
 
 [project.optional-dependencies]
 warp = [
-    "warp-lang==1.9.0",
+    "warp-lang==1.9.1",
 ]
 
 [project.scripts]