From 205f09af2966f3d2636d8e2c84433ae3268aeef8 Mon Sep 17 00:00:00 2001
From: Rory Conlin <roryconlin@gmail.com>
Date: Fri, 20 Oct 2023 19:48:16 -0400
Subject: [PATCH] Update requirements

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index d7f33c5..2236ad8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 equinox
-jax >= 0.3.2, <= 0.4.14
+jax[cpu] >= 0.3.2, <= 0.4.14
 numpy >= 1.20.0, < 1.25.0
 scipy >= 1.5.0, < 1.11.0