Numba 是一个针对 Python 的 即时编译(Just-In-Time, JIT)编译器,专为加速数值计算和科学计算代码设计。它通过将 Python 函数(尤其是涉及 NumPy 数组和循环的代码)转换为高效的机器码,显著提升运行速度,同时保持 Python 代码的简洁性。
核心特点
- 零学习成本:无需编写 C/C++ 扩展或切换语言,仅通过装饰器(如
@njit
)即可标记需要加速的函数。 - 与 NumPy 深度集成:对 NumPy 数组操作有原生优化支持,能高效处理向量化和循环计算。
- 支持 GPU 加速:通过
numba.cuda
模块可直接编写 GPU 并行代码(需 NVIDIA...