Merge pull request #20307 from jakevdp:update-api-tests

PiperOrigin-RevId: 617213115

Author: jax authors

.github/workflows/jax-array-api.yml

@@ -25,7 +25,7 @@ jobs:
       with:
         repository: data-apis/array-api-tests
         # TODO(jakevdp) update this to a stable release/tag when available.
-        ref: '4f83bb3ee9146c30cf997f1527588b7a8b8ee6db'  # Latest commit as of 2024-02-16
+        ref: '58113931e9959ee70dca4bace430b55b90a1eed8'  # Latest commit as of 2024-03-18
         submodules: 'true'
         path: 'array-api-tests'
     - name: Set up Python ${{ matrix.python-version }}