Fix infinite norm negative axis mismatch bug for matrices with 2 or more dimensions.#3756

danlee2002 wants to merge 3 commits into

ml-explore:mainml-explore/mlx:mainfrom

danlee2002:fix-inf-norm-axisdanlee2002/mlx:fix-inf-norm-axisCopy head branch name to clipboard