Collapse the non sorted axes to pick the contiguous sort kernel - #4366

Collapse the non sorted axes to pick the contiguous sort kernel#4366

kapellirohith wants to merge 1 commit into

ml-explore:mainml-explore/mlx:mainfrom

kapellirohith:sort-segment-orderkapellirohith/mlx:sort-segment-orderCopy head branch name to clipboard