Update package versions (#100)

jakevdp · web-flow · commit 5ba7896fc962 · 2024-11-18T09:18:57.000-08:00
- optax==0.2.4
- orbax-checkpoint==0.9.1
- orbax-export==0.0.6
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,9 +18,9 @@ dependencies = [
     "jax==0.4.35",
     "flax==0.10.1",
     "ml_dtypes==0.4.0",
-    "optax==0.2.3",
-    "orbax-checkpoint==0.8.0",
-    "orbax-export==0.0.5",
+    "optax==0.2.4",
+    "orbax-checkpoint==0.9.1",
+    "orbax-export==0.0.6",
 ]
 
 [project.urls]
