diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 56ae9fdd..a7960118 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -37,12 +37,7 @@ jobs:
         pip3 install setuptools
         pip3 install wheel
         pip3 install pytest coverage pytest-cov
-        pip install ./PyAutoConf ./PyAutoArray
-        if [ "${{ matrix.python-version }}" = "3.12" ]; then
-          pip install "./PyAutoArray[optional]"
-        else
-          pip install numba pynufft
-        fi
+        pip install ./PyAutoConf "./PyAutoArray[optional]"
 
     - name: Extract branch name
       shell: bash
diff --git a/pyproject.toml b/pyproject.toml
index 8e0ce01c..3cb33a35 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,7 +8,7 @@ dynamic = ["version"]
   description="PyAuto Data Structures"
 readme = { file = "README.rst", content-type = "text/x-rst" }
 license = { text = "MIT" }
-requires-python = ">=3.12"
+requires-python = ">=3.9"
 authors = [
     { name = "James Nightingale", email = "James.Nightingale@newcastle.ac.uk" },
     { name = "Richard Hayes", email = "richard@rghsoftware.co.uk" },
@@ -18,8 +18,12 @@ classifiers = [
   "Topic :: Scientific/Engineering :: Physics",
   "Natural Language :: English",
   "Operating System :: OS Independent",
+  "Programming Language :: Python :: 3.9",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
-  "Programming Language :: Python :: 3.13"
+  "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14"
 ]
 keywords = ["cli"]
 dependencies = [
@@ -49,7 +53,9 @@ local_scheme = "no-local-version"
 
 
 [project.optional-dependencies]
-optional=[
+jax = ["autoconf[jax]"]
+optional = [
+    "autoarray[jax]",
     "numba",
     "pynufft",
     "tensorflow-probability==0.25.0"