import scipy._external.array_api_compat.numpy as np_compat from scipy._external.array_api_compat import is_array_api_obj, is_jax_array ...