From 1aed7bd7d4bb5bbdf45c3bb71265eec8773d764e Mon Sep 17 00:00:00 2001
From: Justin Chu <justinchu@microsoft.com>
Date: Sun, 5 May 2024 05:52:43 +0000
Subject: [PATCH] jax[cpu]

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

diff --git a/requirements-dev.txt b/requirements-dev.txt
index 0420ccbe6..dfbe51ac2 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -7,7 +7,7 @@ rich>=13.7.1
 
 # Docs site
 furo
-jax
+jax[cpu]
 matplotlib
 ml_dtypes
 myst-parser[linkify]