Test failures on darwin since 0.7.0 #31428
Description
Description
Context: bumping jax from 0.6.2 to 0.7.1 in nixpkgs.
Since the 0.7.0 release, several tests started failing to aarch64-darwin.
The failures are essentially AssertionError on arrays. Interestingly, several report large numerical errors.
Snippet from the failure log:
FAILED tests/linalg_test.py::LaxLinalgTest::testEigh3 - AssertionError: np.float32(14.160136) not less than or equal to np.float32(0.013507687)
FAILED tests/linalg_test.py::LaxLinalgTest::testEigh5 - AssertionError: np.float32(7.058923) not less than or equal to np.float32(0.0075361016)
FAILED tests/linalg_test.py::LaxLinalgTest::testEigh7 - AssertionError: np.float32(9.275328) not less than or equal to np.float32(0.009189508)
========= 74 failed, 29161 passed, 20260 skipped in 1069.20s (0:17:49) =========
Full log available at: https://paste.glepage.com/raw/eel-panda-mouse
System info (python version, jaxlib version, accelerator, etc.)
jax: 0.7.1.dev20250829
jaxlib: 0.7.1
numpy: 2.3.2
python: 3.13.6 (main, Aug 6 2025, 13:05:20) [Clang 19.1.7 ]
device info: cpu-1, 1 local devices"
process_count: 1
platform: uname_result(system='Darwin', node='darwin01', release='24.6.0', version='Darwin Kernel Version 24.6.0: Mon Jul 14 11:30:40 PDT 2025; root:xnu-11417.140.69~1/RELEASE_ARM64_T8132', machine='arm64')
JAX_RELEASE=1
Thanks in advance for your help :)