fix(setup): fix jaxlib version

borisdayma · Mar 9, 2023 · ff6e010 · ff6e010
1 parent b89d6b7
commit ff6e010
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -30,6 +30,7 @@ install_requires =
     emoji
     pillow
     jax==0.3.25
+    jaxlib==0.3.25
     flax==0.6.3
     wandb