pin min version of flax

flaport · Apr 10, 2024 · 8618490 · 8618490
1 parent 54b40cb
commit 8618490
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -27,7 +27,7 @@ classifiers = [
 dependencies = [
   "black",
   "fastcore",
-  "flax",
+  "flax>=0.8.2",
   "jax",
   "jaxlib",
   "jaxtyping",