Numba

optimize-for-gpu 可以根据合适的库选择，帮助把受 CPU 限制的 Python 代码转成 NVIDIA GPU 代码。适用于数组、DataFrame、机器学习流水线、图分析、图像处理、地理空间任务、向量检索和自定义 kernel。它会针对 CuPy、cuDF、cuML、cuGraph、cuCIM、cuVS、KvikIO、Numba CUDA 和 Warp 的选择提供实用的 optimize-for-gpu 用法与迁移建议。