McNUFFT – 基于 MLX 的 Apple Silicon GPU 非均匀快速傅里叶变换

McNUFFT 是一个面向 Apple Silicon GPU 的非均匀快速傅里叶变换（NUFFT）实现，基于 MLX 框架构建。该项目在 GitHub 上开源，旨在为苹果芯片上的科学计算和信号处理提供高效的 GPU 加速方案。

背景速读

- **NUFFT（非均匀快速傅里叶变换）** 是信号处理领域的关键算法，能处理数据点不在规则网格上的场景，广泛用于 MRI 影像重建、天文成像和无线通信。 - **MLX** 是苹果公司 2023 年底开源的深度学习框架，专为 Apple Silicon（M 系列芯片）优化，能充分利用统一内存架构和 GPU，相比 PyTorch/TensorFlow 更高效、内存占用更低。 - **McNUFFT** 是一个基于 MLX 实现的 NUFFT 库，让相关研究人员无需依赖 CUDA（NVIDIA 的 GPU 计算平台）就能在苹果硬件上做高性能 NUFFT 计算——这对 MRI 和信号处理领域的 Mac 用户尤其重要。 - 此前，Apple Silicon 上缺乏成熟的 NUFFT 加速方案，研究者要么用慢得多的 CPU 实现，要么依赖非官方的 Python 绑定。McNUFFT 填补了这一空白。