diff --git a/workspace.bzl b/workspace.bzl
index 284208d280..79b1c8059b 100644
--- a/workspace.bzl
+++ b/workspace.bzl
@@ -1,4 +1,4 @@
-JAX_COMMIT = "0c6398cb866c29b4b6e054117dc156e5dab1f0c2"
+JAX_COMMIT = "9e83c4450db4e51162f86b10a4626e0fffb6fd3d"
 JAX_SHA256 = ""
 
 ENZYME_COMMIT = "53e195b4262f7b208fd6468b5d8f85101d8955a9"