Layer norm throws on dimension mismatch#2870

awni merged 2 commits into

mainml-explore/mlx:mainfrom

layer_norm_throwsml-explore/mlx:layer_norm_throwsCopy head branch name to clipboard