update jax to 0.4.4 in example_project

jorenham · Mar 1, 2023 · b0224c0 · b0224c0
1 parent f89215a
commit b0224c0
Show file tree

Hide file tree

Showing 2 changed files with 98 additions and 132 deletions.
diff --git a/example_project/poetry.lock b/example_project/poetry.lock
diff --git a/example_project/pyproject.toml b/example_project/pyproject.toml
@@ -9,7 +9,7 @@ packages = [{include = "jax_devices_info.py"}]
 
 [tool.poetry.dependencies]
 python = "^3.8"
-jax = {extras = ["cuda"], version = "^0.3.7"}
+jax = {extras = ["cuda"], version = "^0.4.4"}
 
 [[tool.poetry.source]]
 name = "jorenham/jax_pep503"