From 8e40b0f378657e1fe817f0e35a2f8eff0ed4160c Mon Sep 17 00:00:00 2001
From: Justin Guese <guese.justin@gmail.com>
Date: Fri, 24 May 2024 13:50:14 +0200
Subject: [PATCH] dependency

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index 0fb8934..2848739 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -35,6 +35,7 @@ numpy = ">=1.26.4"
 pandas = ">=2.1.4"
 paxml = "1.4.0"
 jax = "0.4.26"
+jaxlib = "0.4.26"
 
 
 [build-system]