Merge pull request #206 from AI-Hypercomputer:wstcliyu/remove-jax-072

PiperOrigin-RevId: 891905545

Author: copybara-github

.github/workflows/test.yml

@@ -24,7 +24,7 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ['3.11', '3.12', '3.13', '3.14']
-        jax-version: ['0.7.2', '0.8.0', '0.8.1', '0.8.2', '0.9.0', 'nightly']
+        jax-version: ['0.8.0', '0.8.1', '0.8.2', '0.9.0', 'nightly']
 
     runs-on: ubuntu-latest
     timeout-minutes: 30

pathwaysutils/__init__.py

@@ -21,4 +21,4 @@
 del _initialize
 
 # When changing this, also update the CHANGELOG.md.
-__version__: str = "v0.1.5"
+__version__ = "v0.1.6"

pyproject.toml

@@ -14,7 +14,7 @@ keywords = []
 dependencies = [
     "absl-py",
     "fastapi",
-    "jax>=0.7.2",
+    "jax>=0.8.0",
     "orbax-checkpoint",
     "uvicorn",
     "requests",