DumPy:告别NumPy的思考,拥抱GPU加速的简洁数组语言
2025-05-24

作者痛斥NumPy在高维数组操作上的繁琐和低效,特别是其复杂的广播和索引机制,导致程序员需要不断思考数组形状和函数行为。为此,作者提出了DumPy,一个简洁的数组语言,通过引入循环和索引的语法,并将其编译成向量化操作,从而实现GPU加速,并避免了NumPy中因避免循环而产生的复杂性。DumPy的核心思想是将维度映射到标签,并利用JAX的`vmap`函数进行向量化计算,从而简化代码,提高效率。作者通过几个例子对比了DumPy、NumPy、JAX和纯循环的性能,结果显示DumPy在简洁性和效率上都表现出色。
开发