Update pyproject.toml

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Oct 23, 2024 · 536bbcd · 536bbcd
1 parent 7e003b2
commit 536bbcd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -139,7 +139,8 @@ jax = [
     'jax>=0.4.33;python_version>="3.10"',
     'flax>=0.8.0;python_version>="3.10"',
     'orbax-checkpoint;python_version>="3.10"',
-    'jax-ai-stack;python_version>="3.10"',
+    # The pinning of ml_dtypes may conflict with TF
+    # 'jax-ai-stack;python_version>="3.10"',
 ]
 
 [tool.deepmd_build_backend.scripts]