[JAX] Update JAX CI dockerfiles to use NumPy 2.0.0rc1, SciPy 1.13.0rc1, and ml_dtypes 0.4.0.

Change in preparation for releasing JAX with NumPy 2.0 support.

PiperOrigin-RevId: 621354875

Author: hawkinsp

.github/workflows/wheel_win_x64.yml

@@ -36,6 +36,8 @@ jobs:
           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.0rc1
           "C:\\msys64\\;C:\\msys64\\usr\\bin\\;" >> $env:GITHUB_PATH
           python.exe build\build.py --bazel_options=--color=yes --verbose
 

.github/workflows/windows_ci.yml

@@ -46,6 +46,8 @@ jobs:
         run: |
           cd jax
           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.0rc1
           "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