Fix nan_to_num replacing inf with 0 for float16 and bfloat16 - #4222

Fix nan_to_num replacing inf with 0 for float16 and bfloat16#4222

zcbenz merged 2 commits into

ml-explore:mainml-explore/mlx:mainfrom

codeAnqiang-ma:fix/nan-to-num-half-infcodeAnqiang-ma/mlx:fix/nan-to-num-half-infCopy head branch name to clipboard