Manuel Villanueva 9cbb1b0148 Modified sort behavior when running CPU or Metal to match NumPy/JAX (#2667) ...

* Modified sort behavior when running CPU or Metal to match NumPy/JAX sorting behavior.

* Modified sort behavior when running CPU or Metal to match NumPy/JAX

* nits

---------

Co-authored-by: Awni Hannun <awni@apple.com>

2025-10-13 14:36:45 -07:00

..

mlx

fix cross entropy axis param (#2641)

2025-10-01 16:49:55 -07:00

scripts

nccl dep + default for cuda (#2526)

2025-08-21 17:57:49 -07:00

src

Export with callback (#2612)

2025-10-08 19:24:33 -07:00

tests

Modified sort behavior when running CPU or Metal to match NumPy/JAX (#2667)

2025-10-13 14:36:45 -07:00