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