I'm opening this issue to track what we need in array-api-extra to unblock work in scipy for JAX adoption:
array-api-extra (not blocked by array-api-compat release):
Not on the critical path:
Shelved indefinitely due to incompatibility with JAX:
CC @rgommers
I'm opening this issue to track what we need in array-api-extra to unblock work in scipy for JAX adoption:
array-api-extra (not blocked by array-api-compat release):
atsupport for bool mask in Dask and JAX #121lazy_apply#86lazywhere#14broadcast_shapes#133atshould not force overwrite in Dask when copy=None #135Not on the critical path:
setdiff1dfor jax.jit and dask NaN-shaped arrays #116 (only used inscipy.optimize.quadratic_assignment, which is numpy-only)Shelved indefinitely due to incompatibility with JAX:
CC @rgommers