mlx - implement `fft2`, `rfft`, and `irfft` and fix `fft`#20781

Merged

fchollet merged 1 commit intokeras-team:mlxfrom acsweet:mlx-math

Jan 18, 2025

+40-8

Commits on Jan 18, 2025

fix to fft, implement fft2, rfft, and irfft for mlx
acsweet
committed