Fix logsumexp JVP axis reduction#3721

chrismicah wants to merge 1 commit into

ml-explore:mainml-explore/mlx:mainfrom

chrismicah:fix/logsumexp-jvp-axischrismicah/mlx:fix/logsumexp-jvp-axisCopy head branch name to clipboard