Add Python 3.13.0 to JAX Docker images with CUDA 12.3 and CUDA 12.1.

Set max Python version to 3.13.0 in JAX Kokoro jobs.

PiperOrigin-RevId: 684216757
This commit is contained in:
jax authors 2024-10-09 16:48:16 -07:00
parent 73418427a8
commit b65be4e1ae
3 changed files with 4 additions and 4 deletions

View File

@ -51,8 +51,8 @@ jobs:
enable-x64: 1
prng-upgrade: 1
num_generated_cases: 1
- name-prefix: "with 3.12"
python-version: "3.12"
- name-prefix: "with 3.13"
python-version: "3.13"
enable-x64: 0
prng-upgrade: 0
num_generated_cases: 1

View File

@ -32,7 +32,7 @@ jobs:
strategy:
fail-fast: false
matrix:
python-version: ["3.12"]
python-version: ["3.13"]
outputs:
artifacts_availability: ${{ steps.status.outputs.ARTIFACTS_AVAILABLE }}
steps:

View File

@ -17,7 +17,7 @@ jobs:
matrix:
os: [windows-2019-32core]
arch: [AMD64]
pyver: ['3.10', '3.11', '3.12', '3.13.0-rc.2']
pyver: ['3.10', '3.11', '3.12', '3.13']
name: ${{ matrix.os }} ${{ matrix.pyver }} jaxlib wheel build
runs-on: ${{ matrix.os }}