optax.

ott-jax · Nov 18, 2024 · 07ceaac · 07ceaac
1 parent 39e56cb
commit 07ceaac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -60,7 +60,7 @@ Changelog = "https://github.com/ott-jax/ott/releases"
 [project.optional-dependencies]
 neural = [
     "flax>=0.6.6",
-    "optax>=0.1.1",
+    "optax>=0.2.4",
     "diffrax>=0.4.1",
 ]
 dev = [