From ff6e010ee0d36449cf63674c2ab1446ba1b82b48 Mon Sep 17 00:00:00 2001
From: Boris Dayma <boris.dayma@gmail.com>
Date: Wed, 8 Mar 2023 21:51:30 -0600
Subject: [PATCH] fix(setup): fix jaxlib version

---
 setup.cfg | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.cfg b/setup.cfg
index 71aa84513..6a3cc6b65 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -30,6 +30,7 @@ install_requires =
     emoji
     pillow
     jax==0.3.25
+    jaxlib==0.3.25
     flax==0.6.3
     wandb