diff --git a/workspace.bzl b/workspace.bzl
index 5412986ab..dd3aaf719 100644
--- a/workspace.bzl
+++ b/workspace.bzl
@@ -1,4 +1,4 @@
-JAX_COMMIT = "3a83fa48995650b659b9200a325b7e551640958b"
+JAX_COMMIT = "a1c8fadb66e372bacf16c93aa6e90fa0aa6ac3af"
 JAX_SHA256 = ""
 
 ENZYME_COMMIT = "e4834164abd7367d75c98ffd90be8c406921376d"