build: bump min python, dependencies

Signed-off-by: Nathaniel Starkman <nstarman@users.noreply.github.com>

.github/workflows/run_tests.yml

@@ -7,7 +7,7 @@ jobs:
   run-test:
     strategy:
       matrix:
-        python-version: [ 3.9, 3.11 ]
+        python-version: [ "3.10", "3.11", "3.13" ]
         os: [ ubuntu-latest ]
       fail-fast: false
     runs-on: ${{ matrix.os }}

pyproject.toml

@@ -3,7 +3,7 @@ name = "quax"
 version = "0.0.5"
 description = "Multiple dispatch in JAX via custom interpreters."
 readme = "README.md"
-requires-python ="~=3.9"
+requires-python ="~=3.10"
 license = {file = "LICENSE"}
 authors = [
   {name = "Patrick Kidger", email = "contact@kidger.site"},
@@ -24,7 +24,7 @@ classifiers = [
 ]
 urls = {repository = "https://github.com/patrick-kidger/quax" }
 dependencies = [
-    "jax>=0.4.11",
+    "jax>=0.4.38",
     "jaxtyping>=0.2.20",
     "equinox>=0.11.0",
     "plum-dispatch>=2.2.1",

tests/requirements.txt

@@ -1,3 +1,3 @@
 beartype
 pytest
-jax[cpu]
+jax[cpu]>=0.4.38