[prady0t]

Reference issue

Closes #1006

What does this implement/fix?

Additional information

AI Generation Disclosure

GitHub Copilot was used to generate changes in _compat.py file

[prady0t]

Most of the changes in these files should be in array-api-compat

[prady0t]

0.32.3 is not yet available on conda-forge. See

[prady0t]

@lucascolley @ev-br Would love some reviews. The gigantic diff is mostly from the pixi lock file.

[ev-br]

For me, a description would be helpful. What is the goal and what is the plan towards it?

[prady0t]

Hi, sorry for the delay. Here's the info:

We want MLX support for the extra library. To do that, we need the tests inside tests/main to pass for MLX. Adding MLX to the list of supported backends causes 900+ failures, most of which are known MLX limitations. Here's a list of all such failures:

#	Failure category	Occurrences	Underlying reason
1	Device support	497 failures + 49 errors	mlx.core.array does not expose .device.
2	astype(copy=...)	97	MLX astype() does not accept the copy keyword.
3	Boolean indexing	56	MLX does not support boolean array indexing.
4	Missing bool attribute	47	MLX exposes bool_, not bool.
5	arange(device=...)	30	MLX arange() does not accept the device keyword.
6	Missing complex128	29	MLX does not expose complex128; it suggests complex64.
7	Zero-dimensional indexing	21	Too many indices are used on a 0-D array.
8	Missing unique_values	14	Function is absent from mlx.core.
9	Missing signbit	11	Function is absent from mlx.core.
10	Sorting API incompatibility	14	MLX sort() and argsort() reject stable and/or axis combinations.
11	Assertion mismatches	14	Dtype, namespace, or expected error-message mismatches.
12	Other missing APIs	2	unique_inverse and __array_api_version__ are missing.

To increase support, I asked AI to create a compatibility layer; that code exists in src/array_api_extra/_lib/_compat.py. Most of these should be an addition to array-api-compat.

Some of the tests (Such as boolean indexing, complex128 dtype tests) are skipped as they are not supported by MLX.