From 7bc47a4c48401026204b6267c92e25a47a77841c Mon Sep 17 00:00:00 2001 From: Luke Baumann Date: Thu, 16 Oct 2025 20:17:37 +0000 Subject: [PATCH] Adding JAX 0.8.0 to the unittest matrix --- .github/workflows/test.yml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 5b18802..638ed43 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -24,7 +24,7 @@ jobs: fail-fast: false matrix: python-version: ['3.10', '3.11', '3.12', '3.13'] - jax-version: ['0.5.3', '0.6.0', '0.6.1', '0.6.2', '0.7.0', '0.7.1', '0.7.2', 'nightly'] + jax-version: ['0.5.3', '0.6.0', '0.6.1', '0.6.2', '0.7.0', '0.7.1', '0.7.2', '0.8.0', 'nightly'] exclude: - python-version: '3.10' @@ -33,6 +33,8 @@ jobs: jax-version: '0.7.1' - python-version: '3.10' jax-version: '0.7.2' + - python-version: '3.10' + jax-version: '0.8.0' - python-version: '3.10' jax-version: 'nightly' runs-on: ubuntu-latest