From 7f9e5659e3b218a6039d249d02dcd6d970d96d96 Mon Sep 17 00:00:00 2001
From: yallup <david.yallup@gmail.com>
Date: Fri, 5 Jul 2024 16:57:04 +0100
Subject: [PATCH] versions

---
 README.md      | 2 +-
 pyproject.toml | 2 --
 2 files changed, 1 insertion(+), 3 deletions(-)

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",