Fix logsumexp jvp to reduce along the axis#3708

obchain wants to merge 1 commit into

ml-explore:mainml-explore/mlx:mainfrom

obchain:fix/logsumexp-jvp-reduceobchain/mlx:fix/logsumexp-jvp-reduceCopy head branch name to clipboard