diff --git a/setup.py b/setup.py index d00a6f5c0..68efbd108 100644 --- a/setup.py +++ b/setup.py @@ -9,7 +9,7 @@ # `pyproject.toml`, once optional dependencies defined there can reference one another extras_require = dict() extras_require["jax"] = [ - "jax[cpu]<0.4.13; platform_system!='Windows'", + "jax[cpu]<0.4.15; platform_system!='Windows'", ] extras_require["zoo"] = [ "tqdm>=4.0",