Skip to content

Pass axis as a Python int in the MLX sort/argsort dispatches - #2400

Open
guillaume-osmo wants to merge 2 commits into
pymc-devs:mainfrom
guillaume-osmo:fix-mlx-sort-axis
Open

guillaume-osmo wants to merge 2 commits into
pymc-devs:mainfrom
guillaume-osmo:fix-mlx-sort-axis

Pass axis as a Python int in the MLX sort/argsort dispatches

11498e7
Select commit
Loading
Failed to load commit list.