Update jax-ml/jax to commit 816e644cc4eb1a4353894ba52d5e0d92716936e5

enzymead-bot[bot] · github-actions[bot] · commit a146de036d18 · 2025-12-26T22:25:47.000Z
Diff: jax-ml/jax@ba024e3...816e644
diff --git a/workspace.bzl b/workspace.bzl
@@ -1,4 +1,4 @@
-JAX_COMMIT = "ba024e35222f9bebadfe7f67395246e11e1eefc5"
+JAX_COMMIT = "816e644cc4eb1a4353894ba52d5e0d92716936e5"
 JAX_SHA256 = ""
 
 ENZYME_COMMIT = "ee83e6901c5032086801e91be9a86f8195883f0d"

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-JAX_COMMIT = "ba024e35222f9bebadfe7f67395246e11e1eefc5"`
	`1`	`+JAX_COMMIT = "816e644cc4eb1a4353894ba52d5e0d92716936e5"`
`2`	`2`	`JAX_SHA256 = ""`
`3`	`3`
`4`	`4`	`ENZYME_COMMIT = "ee83e6901c5032086801e91be9a86f8195883f0d"`