diff --git a/.gitmodules b/.gitmodules
index d8285f3..872dc3a 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -1,3 +1,3 @@
 [submodule "jax"]
 	path = jax
-	url = ../jax.git
+	url = https://github.com/google/jax.git
diff --git a/jax b/jax
index be17053..9b62fd8 160000
--- a/jax
+++ b/jax
@@ -1 +1 @@
-Subproject commit be1705306ca6c0dcce7c7d8bd4fa3ca0a1666f1d
+Subproject commit 9b62fd80d0e6068c0bf05d8fbb184f507b51a087