Test out the wheels job.

.github/workflows/wheel_win_x64.yml

@@ -0,0 +1,62 @@
+name: Wheel build - Windows CPU x86_64
+on:
+  workflow_dispatch: # allows triggering the workflow run manually
+
+env:
+  DISTUTILS_USE_SDK: 1
+  MSSdk: 1
+
+jobs:
+  win-wheels:
+    strategy:
+      fail-fast: false  # Don't stop all wheel builds if one has a test failure.
+      matrix:
+        os: [windows-2019]
+        arch: [AMD64]
+        pyver: ['3.9', '3.10', '3.11', '3.12']
+    name: ${{ matrix.os }} ${{ matrix.pyver }} jaxlib wheel build
+    runs-on: ${{ matrix.os }}
+
+    steps:
+      - name: Cancel Previous Runs
+        uses: styfle/cancel-workflow-action@85880fa0301c86cca9da44039ee3bb12d3bedbfa # ratchet: styfle/[email protected]
+        with:
+          access_token: ${{ github.token }}
+
+      - name: Install LLVM/Clang
+        run: choco install llvm --version=18.1.4 --yes
+
+      - uses: actions/checkout@b4ffde65f46336ab88eb53be808477a3936bae11 # ratchet:actions/checkout@v4
+
+      - uses: actions/setup-python@82c7e631bb3cdc910f68e0081d67478d79c6982d # ratchet:actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.pyver }}
+          cache: 'pip'
+
+      - name: Build wheels
+        env:
+          # TODO(belitskiy): Remove the version once JAX is fully on 6.5.0
+          USE_BAZEL_VERSION: 6.5.0
+          BAZEL_VC: "C:\\Program Files (x86)\\Microsoft Visual Studio\\2019\\Enterprise\\VC"
+          JAXLIB_RELEASE: true
+        run: |
+          python -m pip install -r build/test-requirements.txt
+          python -m pip uninstall -y matplotlib
+          python -m pip install --pre --upgrade numpy==2.0.0rc1 scipy==1.13.0
+          "C:\\msys64\\;C:\\msys64\\usr\\bin\\;" >> $env:GITHUB_PATH
+          python.exe build\build.py `
+            ('--bazel_options=--override_repository=xla=${{ github.workspace }}\xla' -replace '\\','\\') `
+            --bazel_options=--color=yes `
+            --bazel_options=--config=win_clang `
+            --verbose
+
+      - name: Run tests
+        env:
+          JAX_ENABLE_CHECKS: true
+          JAX_SKIP_SLOW_TESTS: true
+          PY_COLORS: 1
+        run: |
+          python -m pip install -e ${{ github.workspace }}
+          python -m pip install --no-index --find-links ${{ github.workspace }}\dist jaxlib
+          echo "JAX_ENABLE_CHECKS=$JAX_ENABLE_CHECKS"
+          pytest -n auto --tb=short tests examples