versions

yallup · Jul 5, 2024 · 7f9e565 · 7f9e565
1 parent b2d75fa
commit 7f9e565
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/README.md b/README.md
@@ -6,7 +6,7 @@ handy classifiers in jax
 
 `pip install .`
 
-For Apple silicon users with Sonoma 14.4+ wanting to try GPU acceleration
+For Apple silicon users with Sonoma 14.4+ wanting to try GPU acceleration (may need `ENABLE_PJRT_COMPATIBILITY=1`)
 
 `pip install ".[metal]"`
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -13,8 +13,6 @@ readme = "README.md"
 license = {file = "LICENSE"}
 requires-python = ">=3.9"
 dependencies = [
-    "jax == 0.4.26",
-    "jaxlib == 0.4.26",
     "flax >= 0.8.2",
     "tqdm >= 4.62.0",
     "optax >= 0.2.2",