fix(nn): apply RoPE on the input shape, not a reshaped 3D tensor - #357

fix(nn): apply RoPE on the input shape, not a reshaped 3D tensor#357

sergey-scherbina wants to merge 1 commit into

oxiglade:mainoxiglade/mlx-rs:mainfrom

sergey-scherbina:fix/rope-decode-single-positionsergey-scherbina/mlx-rs:fix/rope-decode-single-positionCopy head branch name to clipboard