bump flax to v0.10.0 (#70)

jax-ml · Oct 17, 2024 · 8d3faf6 · 8d3faf6
1 parent 1865960
commit 8d3faf6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,7 @@ keywords = []
 # pip dependencies of the project
 dependencies = [
     "jax==0.4.34",
-    "flax==0.9.0",
+    "flax==0.10.0",
     "ml_dtypes==0.4.0",
     "optax==0.2.3",
     "orbax-checkpoint==0.7.0",