Implement batched matmul for large 1D dot products#3580

Ved235 wants to merge 3 commits into

ml-explore:mainml-explore/mlx:mainfrom

Ved235:mainVed235/mlx:mainCopy head branch name to clipboard