Improve batched gemm swizzle#2824

dzoba wants to merge 1 commit intoml-explore:mainml-explore/mlx:mainfrom

dzoba:improve-batched-gemm-swizzledzoba/mlx:improve-batched-gemm-swizzleCopy head branch name to clipboard