From 40f4a3306dd4bd22c9138a5ba88c0b8effb586da Mon Sep 17 00:00:00 2001
From: zombie-einstein <13398815+zombie-einstein@users.noreply.github.com>
Date: Mon, 9 Dec 2024 13:34:37 +0000
Subject: [PATCH] fix: Require jax below 0.4.36 (#269)

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

diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index d09a42d9a..0047cfda2 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -2,7 +2,7 @@ chex>=0.1.3
 dm-env>=1.5
 gymnasium>=1.0
 huggingface-hub
-jax>=0.2.26
+jax>=0.2.26,<0.4.36
 matplotlib~=3.7.4
 numpy>=1.19.5
 Pillow>=9.0.0