feat(compile): CustomKernel and GatherQMM implement output_shapes for shapeless compile#3485

dexwritescode wants to merge 5 commits intoml-explore:mainml-explore/mlx:mainfrom

dexwritescode:fix/custom-kernel-output-shapesdexwritescode/mlx:fix/custom-kernel-output-shapesCopy head branch name to clipboard