NumPy 是 Python 科学计算的基础包，提供高性能 N 维数组、向量化、索引、广播、线性代数、随机数、傅里叶变换和数学函数。

JAX 是面向高性能数组计算和机器学习研究的开源 Python 库，提供自动微分、JIT 编译、自动向量化、并行化、PyTree、随机数和分布式数组能力。