Update jax[cpu] requirement from <0.4.3 to <0.4.4 (#779)

probabilistic-numerics · Feb 13, 2023 · 320462f · 320462f
1 parent 128ce48
commit 320462f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.3; platform_system!='Windows'",
+    "jax[cpu]<0.4.4; platform_system!='Windows'",
 ]
 extras_require["zoo"] = [
     "tqdm>=4.0",