diff --git a/README.md b/README.md
index 54cf711..c8c3ec1 100644
--- 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
index ba5f1e2..3e68753 100644
--- 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",