diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 28a74a0..82d72c5 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -24,9 +24,11 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ['3.11', '3.12', '3.13', '3.14']
-        jax-version: ['0.7.0', '0.7.1', '0.7.2', '0.8.0', '0.8.1', '0.8.2', 'nightly']
+        jax-version: ['0.6.2', '0.7.0', '0.7.1', '0.7.2', '0.8.0', '0.8.1', 'nightly']
 
         exclude:
+          - python-version: '3.14'
+            jax-version: '0.6.2'
           - python-version: '3.14'
             jax-version: '0.7.0'
     runs-on: ubuntu-latest
diff --git a/pyproject.toml b/pyproject.toml
index 5c605da..274c34b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -14,7 +14,7 @@ keywords = []
 dependencies = [
     "absl-py",
     "fastapi",
-    "jax>=0.7.0",
+    "jax>=0.4.26",
     "orbax-checkpoint",
     "uvicorn",
     "requests",
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..1d0cd4a
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,6 @@
+jax[cpu]>=0.5.1
+absl-py
+orbax-checkpoint
+uvicorn
+fastapi
+google-cloud-logging