update pyproject (exclude jax/pax) and update README

2025-04-10 09:47:07 +07:00
parent 373a5293df
commit 6678ba6cee
2 changed files with 12 additions and 6 deletions
@@ -42,7 +42,7 @@ absl-py = ">=1.4.0"

 [tool.poetry.extras]
 pax = ["paxml", "lingvo", "jax", "jaxlib"]
-torch = ["torch",  "jax", "jaxlib"]  # jax & jaxlib are already in pax
+torch = ["torch"] 

 [tool.poetry.dependencies.paxml]
 version = ">=1.4.0"