From c8a6445b0ebe2ac517ed9bee35c62121d3ca88e8 Mon Sep 17 00:00:00 2001 From: diegoferigo Date: Mon, 11 Mar 2024 16:11:19 +0100 Subject: [PATCH] Add again support for latest jax and jaxlib --- setup.cfg | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.cfg b/setup.cfg index 7754644ce..ddc76c923 100644 --- a/setup.cfg +++ b/setup.cfg @@ -53,8 +53,8 @@ package_dir = python_requires = >=3.11 install_requires = coloredlogs - jax >= 0.4.13,< 0.4.25 - jaxlib >= 0.4.13,< 0.4.25 + jax >= 0.4.13 + jaxlib >= 0.4.13 jaxlie >= 1.3.0 jax_dataclasses >= 1.4.0 pptree