Merge pull request #45 from kmheckel/expand-python-314

Add Python 3.14 support

Author: kmheckel

.github/workflows/ci.yml

@@ -62,9 +62,9 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        # pyproject pins requires-python = ">=3.11, <3.14" (flax 0.11 dropped
+        # pyproject pins requires-python = ">=3.11, <3.15" (flax 0.11 dropped
         # 3.10). Keep this aligned with that range.
-        python-version: ["3.11", "3.12", "3.13"]
+        python-version: ["3.11", "3.12", "3.13", "3.14"]
     steps:
       - uses: actions/checkout@v4
       - name: Install uv

AGENTS.md

@@ -203,7 +203,7 @@ forces JAX onto CPU and provides seeded `rngs` / `key` fixtures.
 
 ### Continuous integration
 `.github/workflows/ci.yml` runs four jobs on every PR: `lint` (ruff check +
-format), `test` (pytest matrix over Python 3.11/3.12/3.13 with the quant extra),
+format), `test` (pytest matrix over Python 3.11-3.14 with the quant extra),
 `docs` (`mkdocs build --strict`), and `smoke` (notebook-API drift). Match all
 four locally before pushing. `python-publish.yml` builds and publishes to PyPI
 when a GitHub release is published.
@@ -259,7 +259,7 @@ uv run mkdocs build        # Build static site
 ## Configuration
 
 ### Ruff (`pyproject.toml`)
-- Target: Python 3.11–3.13 (`requires-python = ">=3.11, <3.14"`)
+- Target: Python 3.11–3.14 (`requires-python = ">=3.11, <3.15"`)
 - Line length: 88 (Black-compatible)
 - Enabled rules: Pyflakes (F), pycodestyle errors (E), flake8-bugbear (B), import sorting (I)
 - Excluded: research/, docs/, scripts/, data/

CHANGELOG.md

@@ -32,7 +32,7 @@ modules. This is a **breaking release** — see the
   it over the time axis of `input_data`; `return_all_states=True` also returns
   per-timestep neuron states (membrane traces).
 - **Mixed precision** via `jmp` removed; pass `dtype=` / `param_dtype=` to layers.
-- **Python support** is now `>=3.11, <3.14` (was `>=3.10`).
+- **Python support** is now `>=3.11, <3.15` (was `>=3.10`); `flax>=0.12.7` (Python 3.14 fix) bumps the jax stack to 0.10.x.
 - Dependencies: added `flax>=0.11`, `grain`; removed `dm-haiku`, `jmp`.
 
 ### Added

pyproject.toml

@@ -12,11 +12,12 @@ dependencies = [
     # flax 0.11 changed the nnx.Optimizer signature to require `wrt=` and made
     # `optimizer.update(model, grads)` the canonical call. Spyx's tutorials and
     # spyx.quant target the newer API; pin accordingly.
-    "flax>=0.11.0",
+    # flax >=0.12.7 fixes a Python 3.14 __annotations__ crash in flax.linen.
+    "flax>=0.12.7",
     "grain",
 ]
 readme = "README.md"
-requires-python = ">=3.11, <3.14"
+requires-python = ">=3.11, <3.15"
 license = { text = "MIT" }
 
 [project.optional-dependencies]