fix(mlx): preserve inference norm dtypes#844

Lyxot wants to merge 6 commits into

unslothai:mainunslothai/unsloth-zoo:mainfrom

Lyxot:fix/mlx-norm-output-castLyxot/unsloth-zoo:fix/mlx-norm-output-castCopy head branch name to clipboard