pypi jax

EnzymeAD · Aug 24, 2024 · b2d278b · b2d278b
1 parent d7a3a98
commit b2d278b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/BUILD b/BUILD
@@ -69,7 +69,7 @@ cc_binary(
 py_library(
     name = "enzyme_ad",
     data = [
-        "@jax//jax",
+        "@pypi_jax//:pkg",
         ":enzyme_jax_data",
         "//src/enzyme_ad/jax:enzyme_jax_internal",
     ]