From 97afe3b9eccf32f25cd4eb35a1cf1d508ba80959 Mon Sep 17 00:00:00 2001
From: Samuel Ainsworth <sam.ainsworth@getcruise.com>
Date: Fri, 29 Sep 2023 18:47:05 -0700
Subject: [PATCH] python3Packages.jax: fix build on aarch64-darwin

---
 pkgs/development/python-modules/jax/default.nix | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/pkgs/development/python-modules/jax/default.nix b/pkgs/development/python-modules/jax/default.nix
index a92148ada6bb..dbdf655e4007 100644
--- a/pkgs/development/python-modules/jax/default.nix
+++ b/pkgs/development/python-modules/jax/default.nix
@@ -94,6 +94,14 @@ buildPythonPackage rec {
     "test_for_loop_fixpoint_correctly_identifies_loop_varying_residuals_unrolled_for_loop"
     "testQdwhWithRandomMatrix3"
     "testScanGrad_jit_scan"
+
+    # See https://github.com/google/jax/issues/17867.
+    "test_array"
+    "test_async"
+    "test_copy0"
+    "test_device_put"
+    "test_make_array_from_callback"
+    "test_make_array_from_single_device_arrays"
   ];
 
   disabledTestPaths = lib.optionals (stdenv.isDarwin && stdenv.isAarch64) [