From 7f9e5659e3b218a6039d249d02dcd6d970d96d96 Mon Sep 17 00:00:00 2001 From: yallup 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",