update jax_tpu_embedding version

jshin1394 · jshin1394 · commit 825a1d13c503 · 2025-11-07T00:49:34.000Z
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,7 +43,7 @@ Issues = "https://github.com/google/metrax/issues"
 dev = [
   "absl-py>=2.3.1",
   "jax[cpu]>=0.6.2",
-  "jax_tpu_embedding==0.1.0.dev20250618",
+  "jax_tpu_embedding>=0.1.0.dev20251016",
   "keras-hub",
   "keras-rs>=0.2.1",
   "nltk>=3.9.1",
