From 4b566f62e7548183734e9942f7a46041ddb61a30 Mon Sep 17 00:00:00 2001
From: joj144 <jrdjacoby@gmail.com>
Date: Wed, 19 Feb 2025 09:21:47 -0500
Subject: [PATCH] stopped supporting cuda 11. unpinned scipy and jax

---
 setup.cfg | 13 ++-----------
 1 file changed, 2 insertions(+), 11 deletions(-)

diff --git a/setup.cfg b/setup.cfg
index d9f692e..5398984 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -18,17 +18,8 @@ install_requires =
     optree
 
 [options.extras_require]
-cuda11 = 
-    jax==0.3.22
-    etils==1.5.2
-    scipy==1.11.3
-    chex==0.1.6
-    jaxtyping==0.2.14
-    tensorflow_probability==0.19.0
-
-cuda12 =
-    jax>=0.4.16
-    scipy==1.12.0
+cuda = 
+    jax[cuda12]
 
 [options.package_data]
 * = *.md